Fix TASK_LIST_FORECAST.

DataCanvasIO · Jul 21, 2022 · 5325f5d · 5325f5d
1 parent 9675c3a
commit 5325f5d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hyperts/framework/search_space/micro_search_space.py b/hyperts/framework/search_space/micro_search_space.py
@@ -118,7 +118,7 @@ def __call__(self, *args, **kwargs):
             out = ops.HyperLayer(layers.GlobalAveragePooling1D, name=f'{self.name}_out_gap')(out)
             nas = ops.HyperLayer(layers.Dropout, rate=0.1, name=f'{self.name}_out_drouout')(out)
 
-            if self.task in consts.Task_FORECAST:
+            if self.task in consts.TASK_LIST_FORECAST:
                 default_nas_init_kwargs = self.default_forecasting_init_kwargs
             else:
                 default_nas_init_kwargs = self.default_classification_regression_init_kwargs