Merge pull request #863 from PrefectHQ/couple-more-flakes

mark a couple more flakes

tests/ai/beta/vision/test_cast.py

@@ -112,6 +112,7 @@ def test_map(self):
             Location(city="Washington", state="D.C."),
         )
 
+    @pytest.mark.flaky(reruns=3)
     async def test_async_map(self):
         ny = marvin.beta.Image(
             "https://images.unsplash.com/photo-1568515387631-8b650bbcdb90"

tests/ai/beta/vision/test_extract.py

@@ -57,6 +57,7 @@ def test_ny_image_and_text(self):
             [Location(city="New York City", state="NY")],
         )
 
+    @pytest.mark.flaky(max_runs=3)
     def test_dog(self):
         class Animal(BaseModel, frozen=True):
             type: Literal["cat", "dog", "bird", "frog", "horse", "pig"]

tests/ai/test_cast.py

@@ -95,7 +95,7 @@ def test_cast_text_with_subtle_instructions(self, gpt_4):
 
         def test_str_target_if_only_instructions_provided(self):
             result = marvin.cast(
-                "one", instructions="the numerical representation of the word "
+                "one", instructions="the arabic numeral for the provided word"
             )
             assert isinstance(result, str)
             assert result == "1"