Raw File
test_examples.py
import sys
import pytest

sys.path.append('./examples')


# from train_mnist import ex

@pytest.mark.examples
def test_train_mnist():
    """
    Just check whether the train_mist script still runs...

    NB: This script will fail if you test twice within the same minute, due to
    the name where output files are dumped being the same. Not a problem IRL.
    """
    # TODO: Run this test! Currently this doesn't run on CI, due to a bug in some metadata that TF should be giving
    #       as part of importing itself. This makes it not work with sacred. A PR has been merged into TF, but this
    #       hasn't made it into a release yet. https://github.com/tensorflow/tensorflow/issues/30028
    # ex.run(None, config_updates={"iterations": 2, "hz": {"short": 1}, "float_type": "float32", "jitter_level": 1e-4,
    #                              "num_inducing_points": 100, "dataset": "mnist-small-subset-for-test"})
back to top