fix mypy issue

sotopia-lab · Jan 8, 2025 · 9fe8478 · 9fe8478
1 parent 83cdc4e
commit 9fe8478
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/api/test_fastapi.py b/tests/api/test_fastapi.py
@@ -10,7 +10,7 @@
 from sotopia.messages import SimpleMessage
 from sotopia.api.fastapi_server import app
 import pytest
-from typing import Generator, Callable
+from typing import Generator, Callable, Any
 
 client = TestClient(app)
 
@@ -357,7 +357,7 @@ def test_websocket_simulate(create_mock_data: Callable[[], None]) -> None:
         websocket.send_json(start_msg)
 
         # check the streaming response, stop when we received 2 messages
-        messages = []
+        messages: list[dict[str, Any]] = []
         while len(messages) < 2:
             message = websocket.receive_json()
             assert (