riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/requirements.txt
View File

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

10
examples/test_examples.py
View File

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