From b41ed2f98b551f26fe0d7f8c4e1fb00181130b7d Mon Sep 17 00:00:00 2001
From: Leif Denby <leif@denby.eu>
Date: Tue, 1 Oct 2024 14:30:15 +0000
Subject: [PATCH] remove unused obs_mask arg for vis.plot_prediction

---
 neural_lam/models/ar_model.py | 1 -
 neural_lam/vis.py             | 1 -
 2 files changed, 2 deletions(-)

diff --git a/neural_lam/models/ar_model.py b/neural_lam/models/ar_model.py
index d8d7e99d..b08797e5 100644
--- a/neural_lam/models/ar_model.py
+++ b/neural_lam/models/ar_model.py
@@ -452,7 +452,6 @@ def plot_examples(self, batch, n_examples, prediction=None):
                     vis.plot_prediction(
                         pred=pred_t[:, var_i],
                         target=target_t[:, var_i],
-                        obs_mask=self.interior_mask[:, 0],
                         datastore=self.datastore,
                         title=f"{var_name} ({var_unit}), "
                         f"t={t_i} ({self._datastore.step_length * t_i} h)",
diff --git a/neural_lam/vis.py b/neural_lam/vis.py
index bd991399..4b5fd6a1 100644
--- a/neural_lam/vis.py
+++ b/neural_lam/vis.py
@@ -67,7 +67,6 @@ def plot_error_map(errors, datastore: BaseCartesianDatastore, title=None):
 def plot_prediction(
     pred,
     target,
-    obs_mask,
     datastore: BaseCartesianDatastore,
     title=None,
     vrange=None,