Skip to content

Commit 5325f5d

Browse files
committed
Fix TASK_LIST_FORECAST.
1 parent 9675c3a commit 5325f5d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

hyperts/framework/search_space/micro_search_space.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def __call__(self, *args, **kwargs):
118118
out = ops.HyperLayer(layers.GlobalAveragePooling1D, name=f'{self.name}_out_gap')(out)
119119
nas = ops.HyperLayer(layers.Dropout, rate=0.1, name=f'{self.name}_out_drouout')(out)
120120

121-
if self.task in consts.Task_FORECAST:
121+
if self.task in consts.TASK_LIST_FORECAST:
122122
default_nas_init_kwargs = self.default_forecasting_init_kwargs
123123
else:
124124
default_nas_init_kwargs = self.default_classification_regression_init_kwargs

0 commit comments

Comments
 (0)