SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix example test and test_fetcher for examples (#16478)

Browse Source
This commit is contained in:
Sylvain Gugger
2022-03-29 12:21:19 -04:00
committed by GitHub
parent 86cff21cf6
commit b62ac4d240
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/pytorch/text-classification/run_glue.py
View File

@@ -519,11 +519,11 @@ def main():
if task == "mnli-mm": if task == "mnli-mm":
metrics = {k + "_mm": v for k, v in metrics.items()} metrics = {k + "_mm": v for k, v in metrics.items()}
if "mnli" in task: if task is not None and "mnli" in task:
combined.update(metrics) combined.update(metrics)
trainer.log_metrics("eval", metrics) trainer.log_metrics("eval", metrics)
trainer.save_metrics("eval", combined if "mnli" in task else metrics) trainer.save_metrics("eval", combined if task is not None and "mnli" in task else metrics)
if training_args.do_predict: if training_args.do_predict:
logger.info("*** Predict ***") logger.info("*** Predict ***")

4
utils/tests_fetcher.py
View File

@@ -465,9 +465,9 @@ def infer_tests_to_run(output_file, diff_with_last_commit=False, filters=None):
test_files_to_run.append(f) test_files_to_run.append(f)
# Example files are tested separately # Example files are tested separately
elif f.startswith("examples/pytorch"): elif f.startswith("examples/pytorch"):
test_files_to_run.append("examples/pytorch/test_examples.py") test_files_to_run.append("examples/pytorch/test_pytorch_examples.py")
elif f.startswith("examples/flax"): elif f.startswith("examples/flax"):
test_files_to_run.append("examples/flax/test_examples.py") test_files_to_run.append("examples/flax/test_flax_examples.py")
else: else:
new_tests = module_to_test_file(f) new_tests = module_to_test_file(f)
if new_tests is not None: if new_tests is not None: