From bd1f35e50f1626f81ec6df68a277f640fc8a6e30 Mon Sep 17 00:00:00 2001 From: Weixuan Xiao Date: Thu, 26 Oct 2023 10:37:43 +0200 Subject: [PATCH] Fix WebSocket dataframe arguments creation --- tests/communications/test_websocket_actor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/communications/test_websocket_actor.py b/tests/communications/test_websocket_actor.py index 6fa1831fed..79d6cbef4e 100644 --- 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() ], )