Turn PytestUnhandledCoroutineWarning into an error (mlflow#14399)

Signed-off-by: harupy <[email protected]>
Signed-off-by: Ben Wilson <[email protected]>
Co-authored-by: Ben Wilson <[email protected]>

Author: harupy

pytest.ini

@@ -5,4 +5,7 @@ filterwarnings =
   error:^`np\.[a-z]+` is a deprecated alias for.+:DeprecationWarning:mlflow
   error:^`np\.[a-z]+` is a deprecated alias for.+:DeprecationWarning:tests
   error::pytest.PytestCollectionWarning
+  # TODO: In pytest 8.4.0 (not released yet), this can be removed.
+  # Raise an error if async tests without the `@pytest.mark.asyncio` decorator are found.
+  error::pytest.PytestUnhandledCoroutineWarning
 timeout = 1200

tests/gateway/providers/test_palm.py

@@ -241,22 +241,26 @@ def embeddings_response():
 def embeddings_batch_response():
     return {
         "embeddings": [
-            [
-                3.25,
-                0.7685547,
-                2.65625,
-                -0.30126953,
-                -2.3554688,
-                1.2597656,
-            ],
-            [
-                7.25,
-                0.7685547,
-                4.65625,
-                -0.30126953,
-                -2.3554688,
-                8.2597656,
-            ],
+            {
+                "value": [
+                    3.25,
+                    0.7685547,
+                    2.65625,
+                    -0.30126953,
+                    -2.3554688,
+                    1.2597656,
+                ]
+            },
+            {
+                "value": [
+                    7.25,
+                    0.7685547,
+                    4.65625,
+                    -0.30126953,
+                    -2.3554688,
+                    8.2597656,
+                ]
+            },
         ],
         "headers": {"Content-Type": "application/json"},
     }
@@ -294,6 +298,7 @@ async def test_embeddings(prompt):
         mock_post.assert_called_once()
 
 
+@pytest.mark.asyncio
 async def test_embeddings_batch():
     config = embeddings_config()
     with mock.patch(