From c1182e524aa4ef07c854b47e9d21e16092ae6b07 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Sun, 3 Nov 2019 05:43:30 -0500
Subject: [PATCH] fixing test

---
 tests/test_cpu_models.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/test_cpu_models.py b/tests/test_cpu_models.py
index 62f30404411d7..1b7a2573d70f1 100644
--- a/tests/test_cpu_models.py
+++ b/tests/test_cpu_models.py
@@ -299,6 +299,8 @@ def test_tbptt_cpu_model():
     """
     testing_utils.reset_seed()
 
+    save_dir = testing_utils.init_save_dir()
+
     truncated_bptt_steps = 2
     sequence_size = 30
     batch_size = 30
@@ -366,6 +368,8 @@ def train_dataloader(self):
 
     assert result == 1, 'training failed to complete'
 
+    testing_utils.clear_save_dir()
+
 
 def test_single_gpu_model():
     """