remove unused obs_mask arg for vis.plot_prediction

mllam · Oct 1, 2024 · b41ed2f · b41ed2f
1 parent 2bbe666
commit b41ed2f
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git 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
@@ -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,