cleanup test directory (mlflow#900)

AmeyRuikar · Feb 27, 2019 · 8ab7d38 · 8ab7d38
1 parent 2d6b3ae
commit 8ab7d38
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/store/test_sqlalchemy_store.py b/tests/store/test_sqlalchemy_store.py
@@ -1,3 +1,4 @@
+import shutil
 import unittest
 import warnings
 
@@ -15,7 +16,7 @@
 from mlflow.utils.file_utils import TempDir
 
 DB_URI = 'sqlite://'
-ARTIFACT_URI = 'file://fake file'
+ARTIFACT_URI = 'artifact_folder'
 
 
 class TestSqlAlchemyStoreSqliteInMemory(unittest.TestCase):
@@ -33,6 +34,7 @@ def setUp(self):
     def tearDown(self):
         if self.store:
             models.Base.metadata.drop_all(self.store.engine)
+        shutil.rmtree(ARTIFACT_URI)
 
     def _experiment_factory(self, names):
         if type(names) is list: