fig bug in ensembe_stacked time limit and set version 0.8.8

mljar · Jan 30, 2021 · e2f1569 · e2f1569
1 parent 14e51de
commit e2f1569
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@
 
 setup(
     name="mljar-supervised",
-    version="0.8.7",
+    version="0.8.8",
     description="Automates Machine Learning Pipeline with Feature Engineering and Hyper-Parameters Tuning",
     long_description=long_description,
     long_description_content_type="text/markdown",

diff --git a/supervised/__init__.py b/supervised/__init__.py
@@ -1,3 +1,3 @@
-__version__ = "0.8.7"
+__version__ = "0.8.8"
 
 from supervised.automl import AutoML
diff --git a/supervised/tuner/time_controller.py b/supervised/tuner/time_controller.py
@@ -134,7 +134,8 @@ def compound_time_should_use(self, fit_level):
         return compound
 
     def enough_time_for_step(self, fit_level):
-
+        if fit_level in ["ensemble", "ensemble_stacked"]:
+            return True
         total_time_spend = time.time() - self._start_time
         compound = self.compound_time_should_use(fit_level)
         #print("Enough time for step", fit_level, np.round(total_time_spend,2), np.round(compound,2))