diff --git a/neural_lam/models/ar_model.py b/neural_lam/models/ar_model.py index 0af25367..c875688b 100644 --- a/neural_lam/models/ar_model.py +++ b/neural_lam/models/ar_model.py @@ -534,10 +534,6 @@ def plot_examples(self, batch, n_examples, split, prediction=None): ] example_i = self.plotted_examples - # for i, fig in enumerate(var_figs): - # fn = f"example_{i}_{example_i}_t{t_i}.png" - # fig.savefig(fn) - # logger.info(f"Saved example plot to {fn}") wandb.log( {