fix test

PrefectHQ · Apr 18, 2024 · 0dec50a · 0dec50a
1 parent b86ec9f
commit 0dec50a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/tests/ai/beta/vision/test_cast.py b/tests/ai/beta/vision/test_cast.py
@@ -13,8 +13,7 @@ def assert_locations_equal(observed: Location, expected: Location):
     assert_equal(
         observed,
         expected,
-        instructions="Do the locations refer to the same place even if the literal words differ? NYC, New York, and New York City are all the same place.",
-        model="gpt-3.5-turbo",
+        instructions="The location models may not be literally identical, but do they refer to the same city?",
     )
 
 

diff --git a/tests/ai/beta/vision/test_extract.py b/tests/ai/beta/vision/test_extract.py
@@ -15,8 +15,7 @@ def assert_locations_equal(observed: Location, expected: Location):
     assert_equal(
         observed,
         expected,
-        instructions="Do the locations refer to the same place even if the literal words differ? NYC, New York, and New York City are all the same place.",
-        model="gpt-3.5-turbo",
+        instructions="The location models may not be literally identical, but do they refer to the same city?",
     )