From a374a7ea00b1ca384413404642a60f71427c31dd Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Fri, 26 Jul 2019 12:33:35 -0400
Subject: [PATCH] added saving tests to cpu

---
 tests/test_models.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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