diff --git a/autoPyTorch/datasets/time_series_dataset.py b/autoPyTorch/datasets/time_series_dataset.py index fe5ddd234..670eb44c9 100644 --- a/autoPyTorch/datasets/time_series_dataset.py +++ b/autoPyTorch/datasets/time_series_dataset.py @@ -81,9 +81,7 @@ def compute_time_features(start_time: pd.DatetimeIndex, freq=freq)[-time_feature_length:] try: time_features = np.vstack( - [transform(date_info).to_numpy(float) - if not isinstance(transform, ConstantTransform) else transform(date_info) - for transform in time_feature_transforms] + [transform(date_info) for transform in time_feature_transforms] ).T except OutOfBoundsDatetime: # This is only a temporal solution TODO consider how to solve this! diff --git a/setup.py b/setup.py index fff10a9d5..4198ff6b5 100755 --- a/setup.py +++ b/setup.py @@ -49,7 +49,7 @@ include_package_data=True, extras_require={ "forecasting": [ - "gluonts", + "gluonts>=0.10.0", "sktime", "pytorch-forecasting", ],