riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-15 10:38:23 +06:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Refactor the function get_results (#20999)

Browse Source
This commit is contained in:
milyiyo 2023-01-04 12:05:36 -05:00 committed by GitHub
parent 926452298d
commit 3b309818e7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
examples/flax/test_flax_examples.py
View File

@ -60,14 +60,11 @@ def get_setup_file():
def get_results(output_dir, split="eval"): def get_results(output_dir, split="eval"):
results = {}
path = os.path.join(output_dir, f"{split}_results.json") path = os.path.join(output_dir, f"{split}_results.json")
if os.path.exists(path): if os.path.exists(path):
with open(path, "r") as f: with open(path, "r") as f:
results = json.load(f) return json.load(f)
else:
raise ValueError(f"can't find {path}") raise ValueError(f"can't find {path}")
return results
stream_handler = logging.StreamHandler(sys.stdout) stream_handler = logging.StreamHandler(sys.stdout)