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

[examples testing] restore code (#7099)

Browse Source 
For some reason https://github.com/huggingface/transformers/pull/5512 re-added temp dir creation code that was removed by
https://github.com/huggingface/transformers/pull/6494 defeating the purpose of that PR for those tests.
This commit is contained in:
Stas Bekman
2020-09-14 05:54:23 -07:00
committed by GitHub
parent 4d39148419
commit 3ca1874ca4
2 changed files with 3 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/requirements.txt
View File

@@ -15,4 +15,4 @@ pandas
datasets datasets
fire fire
pytest pytest
conllu conllu

10
examples/test_examples.py
View File

@@ -80,10 +80,7 @@ class ExamplesTests(TestCasePlus):
--warmup_steps=2 --warmup_steps=2
--seed=42 --seed=42
--max_seq_length=128 --max_seq_length=128
""" """.split()
output_dir = "./tests/fixtures/tests_samples/temp_dir_{}".format(hash(testargs))
testargs += "--output_dir " + output_dir
testargs = testargs.split()
if is_cuda_and_apex_available(): if is_cuda_and_apex_available():
testargs.append("--fp16") testargs.append("--fp16")
@@ -149,10 +146,7 @@ class ExamplesTests(TestCasePlus):
--do_train --do_train
--do_eval --do_eval
--num_train_epochs=1 --num_train_epochs=1
""" """.split()
output_dir = "./tests/fixtures/tests_samples/temp_dir_{}".format(hash(testargs))
testargs += "--output_dir " + output_dir
testargs = testargs.split()
if torch_device != "cuda": if torch_device != "cuda":
testargs.append("--no_cuda") testargs.append("--no_cuda")