From df4d39cb5bb554510cef7cb15d5ed3e748c5795c Mon Sep 17 00:00:00 2001
From: Leif Denby <leif@denby.eu>
Date: Wed, 13 Nov 2024 18:18:39 +0100
Subject: [PATCH] fix typo (missing datastore) in ARModel.on_test_epoch_end

---
 neural_lam/models/ar_model.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/neural_lam/models/ar_model.py b/neural_lam/models/ar_model.py
index 3f9729a..bc4c671 100644
--- a/neural_lam/models/ar_model.py
+++ b/neural_lam/models/ar_model.py
@@ -601,7 +601,8 @@ def on_test_epoch_end(self):
                 vis.plot_spatial_error(
                     error=loss_map,
                     datastore=self._datastore,
-                    title=f"Test loss, t={t_i} ({self.step_length * t_i} h)",
+                    title=f"Test loss, t={t_i} "
+                    f"({self._datastore.step_length * t_i} h)",
                 )
                 for t_i, loss_map in zip(
                     self.args.val_steps_to_log, mean_spatial_loss