diff --git a/tests/test_models.py b/tests/test_models.py index ff5b4037a0..f2faad0d5b 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -27,6 +27,8 @@ np.random.seed(SEED) def test_loading_meta_tags(): hparams = get_hparams() + save_dir = init_save_dir() + # save tags exp = get_exp(False) exp.tag({'some_str':'a_str', 'an_int': 1, 'a_float': 2.0}) @@ -41,7 +43,6 @@ def test_loading_meta_tags(): clear_save_dir() - def test_dp_output_reduce(): # test identity when we have a single gpu