Break up test

Arize-ai · Sep 17, 2024 · cb50319 · cb50319
1 parent 0fd14ff
commit cb50319
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packages/phoenix-evals/tests/phoenix/evals/functions/test_classify.py b/packages/phoenix-evals/tests/phoenix/evals/functions/test_classify.py
@@ -368,7 +368,8 @@ def test_llm_classify_prints_to_stdout_with_verbose_flag(
     assert "Snapped '\\nrelevant ' to rail: relevant" in out, "Snapping events should be printed"
     assert "Cannot snap 'unparsable' to rails" in out, "Snapping events should be printed"
     assert "OpenAI invocation parameters" in out, "Model-specific information should be printed"
-    assert "'model': 'gpt-4', 'temperature': 0.0" in out, "Model information should be printed"
+    assert "'model': 'gpt-4'" in out, "Model information should be printed"
+    assert "'temperature': 0.0" in out, "Model information should be printed"
     assert "sk-0123456789" not in out, "Credentials should not be printed out in cleartext"