diff --git a/pelita/ipython/__init__.py b/pelita/ipython/__init__.py index d240a0664..3e5a514df 100644 --- a/pelita/ipython/__init__.py +++ b/pelita/ipython/__init__.py @@ -11,13 +11,10 @@ import zmq from zmq.eventloop.zmqstream import ZMQStream -from IPython.core.display import HTML import os -from ..messaging.json_convert import json_converter -from ..datamodel import Wall, Food - import json +import pelita class ZMQPubSub: def __init__(self, callback): @@ -64,11 +61,11 @@ def on_close(self): print("WebSocket closed") def on_data(self, data): - msg_objs = json_converter.loads(data[0].decode()) - - data = msg_objs.get("__data__") or {} + py_obj = json.loads(data[0].decode()) + data = py_obj.get("__data__") or {} universe = data.get("universe") + universe = pelita.datamodel.CTFUniverse._from_json_dict(universe) game_state = data.get("game_state") if universe: