fix the prediction files

henryaddison · Dec 14, 2023 · 708ce08 · 708ce08
1 parent 275c988
commit 708ce08
Showing 1 changed file with 17 additions and 17 deletions.
diff --git a/bin/check_preds.py b/bin/check_preds.py
@@ -67,20 +67,20 @@ def check(pred_ds):
                     print(e)
             pbar.update(1)
 
-# with logging_redirect_tqdm():
-#     with tqdm(
-#         total=len(fpaths),
-#         desc=f"Fixing prediction files",
-#         unit=" files",
-#     ) as pbar:
-#         for fpath in fpaths:
-#             pred_ds = xr.open_dataset(fpath)
-#             # import pdb; pdb.set_trace()
-#             pred_ds = fix(pred_ds)
-#             errors = check(pred_ds)
-
-#             if len(errors) > 0:
-#                 print(f"Errors in {fpath}:")
-#                 for e in errors:
-#                     print(e)
-#             pbar.update(1)
+with logging_redirect_tqdm():
+    with tqdm(
+        total=len(fpaths),
+        desc=f"Fixing prediction files",
+        unit=" files",
+    ) as pbar:
+        for fpath in fpaths:
+            pred_ds = xr.open_dataset(fpath)
+            # import pdb; pdb.set_trace()
+            pred_ds = fix(pred_ds)
+            errors = check(pred_ds)
+
+            if len(errors) > 0:
+                print(f"Errors in {fpath}:")
+                for e in errors:
+                    print(e)
+            pbar.update(1)