This commit is contained in:
Ines Montani 2020-12-09 11:23:07 +11:00
parent 758ad6c3cd
commit febf71af28
1 changed files with 1 additions and 1 deletions

View File

@ -368,7 +368,7 @@ def test_parse_cli_overrides():
@pytest.mark.parametrize("optimize", ["efficiency", "accuracy"]) @pytest.mark.parametrize("optimize", ["efficiency", "accuracy"])
def test_init_config(lang, pipeline, optimize): def test_init_config(lang, pipeline, optimize):
# TODO: add more tests and also check for GPU with transformers # TODO: add more tests and also check for GPU with transformers
init_config("-", lang=lang, pipeline=pipeline, optimize=optimize, cpu=True) init_config("-", lang=lang, pipeline=pipeline, optimize=optimize, gpu=False)
def test_model_recommendations(): def test_model_recommendations():