wait before expt delete

Author: pplonski

tests/experiment_client_test.py

@@ -4,6 +4,7 @@
 import os
 import unittest
 import pandas as pd
+import time
 
 from mljar.client.project import ProjectClient
 from mljar.client.dataset import DatasetClient
@@ -38,9 +39,12 @@ def setUp(self):
 
 
     def tearDown(self):
+        # wait before clean, to have time to initialize models
+        time.sleep(60)
         # clean
         self.project_client.delete_project(self.project.hid)
 
+
     def test_create_with_kfold_cv(self):
         """
         Create experiment test with k-fold CV.
@@ -161,3 +165,7 @@ def test_create_if_exists(self):
         self.assertEqual(experiment_2.metric, experiment.metric)
         self.assertEqual(experiment_2.validation_scheme, experiment.validation_scheme)
         self.assertTrue(experiment.equal(experiment_2))
+
+
+if __name__ == "__main__":
+    unittest.main()