Fix WebSocket dataframe arguments creation

Giskard-AI · Oct 26, 2023 · bd1f35e · bd1f35e
1 parent a92296f
commit bd1f35e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/communications/test_websocket_actor.py b/tests/communications/test_websocket_actor.py
@@ -155,7 +155,7 @@ def test_websocket_actor_run_model_for_data_frame_regression_internal(request):
     dataframe = websocket.DataFrame(
         rows=[
             websocket.DataRow(columns={
-                k: v for k, v in row.items()
+                str(k): str(v) for k, v in row.items()
             }) for _, row in dataset.df.iterrows()
         ],
     )
@@ -191,7 +191,7 @@ def test_websocket_actor_run_model_for_data_frame_classification_internal(reques
     dataframe = websocket.DataFrame(
         rows=[
             websocket.DataRow(columns={
-                k: v for k, v in row.items()
+                str(k): str(v) for k, v in row.items()
             }) for _, row in dataset.df.iterrows()
         ],
     )