Merge pull request Kaggle#373 from Kaggle/add-talos

close h2o session in test
AlouiAmine · Nov 13, 2018 · a0a46be · a0a46be
2 parents 3fc7e45 + 0020fba
commit a0a46be
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_h2o.py b/tests/test_h2o.py
@@ -3,6 +3,9 @@
 import h2o
 
 class TestH2o(unittest.TestCase):
+    def tearDown(self):
+        h2o.cluster().shutdown(False)
+
     def test_init_read(self):
         h2o.init()
         train = h2o.import_file("/input/tests/data/train.csv", destination_frame="train")