From d45867732c7fbdec2bb1d06c316f80de98576b1d Mon Sep 17 00:00:00 2001 From: Leif Denby Date: Wed, 13 Nov 2024 16:19:53 +0100 Subject: [PATCH] fix bug for datastore ref in ARModel.plot_examples() --- neural_lam/models/ar_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/neural_lam/models/ar_model.py b/neural_lam/models/ar_model.py index 21a28f3..3f9729a 100644 --- 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,