save the fixed prediction files too

henryaddison · Dec 14, 2023 · a08c581 · a08c581
1 parent 708ce08
commit a08c581
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/bin/check_preds.py b/bin/check_preds.py
@@ -74,7 +74,7 @@ def check(pred_ds):
         unit=" files",
     ) as pbar:
         for fpath in fpaths:
-            pred_ds = xr.open_dataset(fpath)
+            pred_ds = xr.load_dataset(fpath)
             # import pdb; pdb.set_trace()
             pred_ds = fix(pred_ds)
             errors = check(pred_ds)
@@ -83,4 +83,6 @@ def check(pred_ds):
                 print(f"Errors in {fpath}:")
                 for e in errors:
                     print(e)
+            else:
+                pred_ds.to_netcdf(fpath)
             pbar.update(1)