Raise error if calling deepsensor.plot.prediction with forecast

alan-turing-institute · Oct 20, 2024 · fab1fd7 · fab1fd7
1 parent 69a6097
commit fab1fd7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/deepsensor/plot.py b/deepsensor/plot.py
@@ -954,6 +954,8 @@ def prediction(
             ax = axes[row_i, col_i]
 
             if pred.mode == "on-grid":
+                if "init_time" in pred[0].indexes:
+                    raise ValueError("Plotting forecasts not currently supported.")
                 if param == "std":
                     vmin = 0
                 else:
@@ -1000,6 +1002,8 @@ def prediction(
                     # )
 
             elif pred.mode == "off-grid":
+                if "init_time" in pred[0].index.names:
+                    raise ValueError("Plotting forecasts not currently supported.")
                 import seaborn as sns
 
                 hue = (