From 07013ac2bd3ef99a21dc808a5313f2daccac4245 Mon Sep 17 00:00:00 2001
From: Jeremiah Lowin <153965+jlowin@users.noreply.github.com>
Date: Mon, 17 Jun 2024 14:58:38 -0400
Subject: [PATCH] Fix xfail marker

---
 tests/ai_tests/test_tasks.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/ai_tests/test_tasks.py b/tests/ai_tests/test_tasks.py
index 35546a45..b07d21bb 100644
--- 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',