Fix xfail marker

PrefectHQ · Jun 17, 2024 · 07013ac · 07013ac
1 parent 134ec38
commit 07013ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/ai_tests/test_tasks.py b/tests/ai_tests/test_tasks.py
@@ -21,7 +21,7 @@ def test_task_pydantic_result(self):
         assert isinstance(result, Name)
         assert result == Name(first="John", last="Doe")
 
-    @pytest.xfail(reason="Need to revisit dataframe handling")
+    @pytest.mark.xfail(reason="Need to revisit dataframe handling")
     def test_task_dataframe_result(self):
         task = Task(
             'return a dataframe with column "x" that has values 1 and 2 and column "y" that has values 3 and 4',