skip cv preparation if eval_method is holdout (#314)

* skip cv preparation if eval_method is holdout * bump version to 0.8.1
microsoft · Nov 28, 2021 · 1545d5a · 1545d5a
1 parent 2f25a87
commit 1545d5a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/flaml/automl.py b/flaml/automl.py
@@ -1128,6 +1128,9 @@ def _prepare_data(self, eval_method, split_ratio, n_splits):
         self._state.X_val, self._state.y_val = X_val, y_val
         self._state.X_train_all = X_train_all
         self._state.y_train_all = y_train_all
+        if eval_method == "holdout":
+            self._state.kf = None
+            return
         if self._split_type == "group":
             # logger.info("Using GroupKFold")
             assert (

diff --git a/flaml/version.py b/flaml/version.py
@@ -1 +1 @@
-__version__ = "0.8.0"
+__version__ = "0.8.1"