From e55e35adea03993de87b23f092b14c6af623d487 Mon Sep 17 00:00:00 2001 From: Mark Harley Date: Sun, 19 Feb 2023 00:38:54 +0000 Subject: [PATCH] Revert c6a5dd1a0 --- flaml/automl/model.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/flaml/automl/model.py b/flaml/automl/model.py index 08d40173a..4df09294d 100644 --- a/flaml/automl/model.py +++ b/flaml/automl/model.py @@ -571,7 +571,9 @@ class TransformersEstimator(BaseEstimator): @property def data_collator(self): from flaml.automl.task.task import Task - from flaml.automl.nlp.huggingface.data_collator import task_to_datacollator_class + from flaml.automl.nlp.huggingface.data_collator import ( + task_to_datacollator_class, + ) data_collator_class = task_to_datacollator_class.get( self._task.name if isinstance(self._task, Task) else self._task @@ -2040,7 +2042,7 @@ class TS_SKLearn(SKLearnEstimator): space.update( { "optimize_for_horizon": { - "domain": tune.choice([False]), + "domain": tune.choice([True, False]), "init_value": False, "low_cost_init_value": False, },