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

…ntel-analytics#5483) When the input data to AutoformerForecaster.predict() is incorrect, the error message is confusing. Thus, a new error message is added.
ForJadeForest · Sep 20, 2022 · d6ab2ba · d6ab2ba
1 parent 3dc4985
commit d6ab2ba
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]),