fix: remove missing import

instructor-ai · Nov 21, 2024 · c2924c4 · c2924c4
1 parent 8652b9c
commit c2924c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/llm/test_writer/evals/test_sentiment_analysis.py b/tests/llm/test_writer/evals/test_sentiment_analysis.py
@@ -36,7 +36,7 @@ class SentimentAnalysis(BaseModel):
 
 @pytest.mark.parametrize("model, data, mode", product(models, test_data, modes))
 def test_writer_sentiment_analysis(
-    model: str, data: List[Tuple], mode: instructor.Mode
+    model: str, data: list[tuple[str, Sentiment]], mode: instructor.Mode
 ):
     client = instructor.from_writer(client=Writer(), mode=mode)