Added more error handling

450f5236 · Brian Loyal · 95c83ae1 · 450f5236
Commit 450f5236 authored Aug 12, 2022 by Brian Loyal
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

run_pretrained_openfold.py run_pretrained_openfold.py +4 -1

No files found.
--- a/run_pretrained_openfold.py
+++ b/run_pretrained_openfold.py
@@ -509,8 +509,11 @@ def update_timings(dict, output_file=os.path.join(os.getcwd(), "timings.json")):
    import json
    if os.path.exists(output_file):
        with open(output_file, "r") as f:
-            if f:
+            try:
                timings = json.load(f)
+            except json.JSONDecodeError:
+                logger.info(f"Overwriting non-standard JSON in {output_file}.")
+                timings = {}
    else:
        timings = {}
    timings.update(dict)