This commit is contained in:
Ines Montani 2020-12-09 11:24:01 +11:00
parent 5d605d539d
commit b85bd63eca
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
config = init_config("-", lang=lang, pipeline=pipeline, optimize=optimize, cpu=True) config = init_config(lang=lang, pipeline=pipeline, optimize=optimize, cpu=True)
assert isinstance(config, Config) assert isinstance(config, Config)