Update tests

PrefectHQ · May 14, 2024 · cf6aafb · cf6aafb
1 parent 55a7848
commit cf6aafb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tests/ai/test_classify.py b/tests/ai/test_classify.py
@@ -123,6 +123,7 @@ def test_return_index(self):
             assert result == 1
 
     class TestExamples:
+        @pytest.mark.flaky(max_runs=3)
         async def test_hogwarts_sorting_hat(self):
             description = "Brave, daring, chivalrous, and sometimes a bit reckless -- just like Harry Potter."
 

diff --git a/tests/ai/vision/test_classify.py b/tests/ai/vision/test_classify.py
@@ -21,8 +21,7 @@ def test_ny_image_and_text(self):
             "https://images.unsplash.com/photo-1568515387631-8b650bbcdb90"
         )
         result = marvin.classify(
-            data="I see the empire state building",
-            images=[img],
+            data=["I see a tall building", img],
             labels=["urban", "rural"],
         )
         assert result == "urban"

diff --git a/tests/ai/vision/test_extract.py b/tests/ai/vision/test_extract.py
@@ -82,7 +82,7 @@ def test_dog_breeds_with_text(self):
             "https://images.unsplash.com/photo-1548199973-03cce0bbc87b?q=80&w=2969&auto=format&fit=crop&ixlib=rb-4.0.3&ixid=M3wxMjA3fDB8MHxwaG90by1wYWdlfHx8fGVufDB8fHx8fA%3D%3D"
         )
         result = marvin.extract(
-            [img, "collie"],
+            [img, "Another type is: collie"],
             target=str,
             instructions="extract all dog breeds pictured or mentioned",
         )