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

[testing] replace hardcoded paths to allow running tests from anywhere (#6523)

Browse Source 
* [testing] replace hardcoded paths to allow running tests from anywhere

* fix the merge conflict
This commit is contained in:
Stas Bekman
2020-08-27 09:22:18 -07:00
committed by GitHub
parent 9d1b4db2aa
commit e6b811f0a7
3 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_trainer.py
View File

@@ -4,7 +4,7 @@ import nlp
import numpy as np
from transformers import AutoTokenizer, TrainingArguments, is_torch_available
from transformers.testing_utils import require_torch
from transformers.testing_utils import get_tests_dir, require_torch
if is_torch_available():
@@ -20,7 +20,7 @@ if is_torch_available():
)
PATH_SAMPLE_TEXT = "./tests/fixtures/sample_text.txt"
PATH_SAMPLE_TEXT = f"{get_tests_dir()}/fixtures/sample_text.txt"
class RegressionDataset:
@@ -262,7 +262,7 @@ class TrainerIntegrationTest(unittest.TestCase):
tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
model = AutoModelForSequenceClassification.from_pretrained(MODEL_ID)
data_args = GlueDataTrainingArguments(
task_name="mrpc", data_dir="./tests/fixtures/tests_samples/MRPC", overwrite_cache=True
task_name="mrpc", data_dir=f"{get_tests_dir()}/fixtures/tests_samples/MRPC", overwrite_cache=True
)
eval_dataset = GlueDataset(data_args, tokenizer=tokenizer, mode="dev")