fix: drop enums for anthropic

instructor-ai · Apr 29, 2024 · 4fc44c9 · 4fc44c9
1 parent 973d4ae
commit 4fc44c9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/llm/test_anthropic/evals/test_simple.py b/tests/llm/test_anthropic/evals/test_simple.py
@@ -98,6 +98,7 @@ class User(BaseModel):
         assert isinstance(member, str)
 
 
+@pytest.mark.skip("Just use Literal!")
 def test_enum():
     class Role(str, Enum):
         ADMIN = "admin"
@@ -110,15 +111,15 @@ class User(BaseModel):
     resp = client.messages.create(
         model="claude-3-haiku-20240307",
         max_tokens=1024,
-        max_retries=0,
+        max_retries=1,
         messages=[
             {
                 "role": "user",
                 "content": "Create a user for a model with a name and role of admin.",
             }
         ],
         response_model=User,
-    )  # type: ignore
+    )
 
     assert isinstance(resp, User)
     assert resp.role == Role.ADMIN