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"])
def test_init_config(lang, pipeline, optimize):
# 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():