Use GPT-4 for sharding spancat test case.

explosion · Apr 20, 2024 · 7653a7b · 7653a7b
1 parent 174ef84
commit 7653a7b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/spacy_llm/tests/sharding/test_sharding.py b/spacy_llm/tests/sharding/test_sharding.py
@@ -172,6 +172,9 @@ def test_sharding_sentiment(config):
 @pytest.mark.skipif(has_openai_key is False, reason="OpenAI API key not available")
 def test_sharding_spancat(config):
     context_length = 265
+    config["components"]["llm"]["model"]["@llm_models"] = "spacy.OpenAI.v1"
+    # Spancat (not sharding) aspect of test case doesn't work with gpt-3.5.
+    config["components"]["llm"]["model"]["name"] = "gpt-4"
     config["components"]["llm"]["model"]["context_length"] = context_length
     config["components"]["llm"]["task"] = {
         "@llm_tasks": "spacy.SpanCat.v3",