Skip to content

GitLab

Commit 95c83ae1 authored by Brian Loyal's avatar Brian Loyal
Browse files

Bug fix

parent 32dd89bb
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
run_pretrained_openfold.py
View file @ 95c83ae1
...@@ -509,12 +509,13 @@ def update_timings(dict, output_file=os.path.join(os.getcwd(), "timings.json")): ...@@ -509,12 +509,13 @@ def update_timings(dict, output_file=os.path.join(os.getcwd(), "timings.json")):
import json import json
if os.path.exists(output_file): if os.path.exists(output_file):
with open(output_file, "r") as f: with open(output_file, "r") as f:
timings = json.loads(f) if f:
timings = json.load(f)
else: else:
timings = {} timings = {}
timings.update(dict) timings.update(dict)
with open(output_file, "w") as f: with open(output_file, "w") as f:
timings.dumps(f) json.dump(timings, f)
return output_file return output_file
if __name__ == "__main__": if __name__ == "__main__":
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment