Chronos: an error message is added to AutoformerForecaster.predict() (#…

…5483) When the input data to AutoformerForecaster.predict() is incorrect, the error message is confusing. Thus, a new error message is added.
intel-analytics · Aug 23, 2022 · 9287e0b · 9287e0b
1 parent 6b9d090
commit 9287e0b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/chronos/src/bigdl/chronos/forecaster/autoformer_forecaster.py b/python/chronos/src/bigdl/chronos/forecaster/autoformer_forecaster.py
@@ -368,6 +368,9 @@ def predict(self, data, batch_size=32):
         """
         if self.distributed:
             invalidInputError(False, "distributed is not support in Autoformer")
+        invalidInputError(isinstance(data, tuple) or isinstance(data, DataLoader),
+                          "The input data to predict() support formats: numpy ndarray tuple"
+                          f" and pytorch dataloader, but found {type(data)}.")
         if isinstance(data, tuple):
             data = DataLoader(TensorDataset(torch.from_numpy(data[0]),
                                             torch.from_numpy(data[1]),