From febf71af280c62dd93460ce1fdcb14cd5824bd10 Mon Sep 17 00:00:00 2001 From: Ines Montani Date: Wed, 9 Dec 2020 11:23:07 +1100 Subject: [PATCH] Fix test --- spacy/tests/test_cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/tests/test_cli.py b/spacy/tests/test_cli.py index 62584d0ce..1f5cd1972 100644 --- a/spacy/tests/test_cli.py +++ b/spacy/tests/test_cli.py @@ -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():