fix bug for datastore ref in ARModel.plot_examples()

mllam · Nov 13, 2024 · d458677 · d458677
1 parent 258079c
commit d458677
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neural_lam/models/ar_model.py b/neural_lam/models/ar_model.py
@@ -461,7 +461,7 @@ def plot_examples(self, batch, n_examples, prediction=None):
                     vis.plot_prediction(
                         pred=pred_t[:, var_i],
                         target=target_t[:, var_i],
-                        datastore=self.datastore,
+                        datastore=self._datastore,
                         title=f"{var_name} ({var_unit}), "
                         f"t={t_i} ({self._datastore.step_length * t_i} h)",
                         vrange=var_vrange,