fit updates in gluonts (#445)

dengdifan · web-flow · commit 95c5fe4832bf · 2022-07-04T18:08:03.000+02:00
* fit updates in gluonts

* fit gluonts version
diff --git 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
@@ -49,7 +49,7 @@
     include_package_data=True,
     extras_require={
         "forecasting": [
-            "gluonts",
+            "gluonts>=0.10.0",
             "sktime",
             "pytorch-forecasting",
         ],
