diff --git a/bec_widgets/cli/client_utils.py b/bec_widgets/cli/client_utils.py index e7803f32..eab0b442 100644 --- a/bec_widgets/cli/client_utils.py +++ b/bec_widgets/cli/client_utils.py @@ -250,7 +250,6 @@ class BECGuiClient(RPCBase): self._client.connector.register( MessageEndpoints.gui_registry_state(self._gui_id), cb=self._handle_registry_update, - parent=self, from_start=True, ) diff --git a/bec_widgets/cli/rpc/rpc_base.py b/bec_widgets/cli/rpc/rpc_base.py index 5383c8c0..deceb383 100644 --- a/bec_widgets/cli/rpc/rpc_base.py +++ b/bec_widgets/cli/rpc/rpc_base.py @@ -248,9 +248,7 @@ class RPCBase: self._rpc_response = None self._msg_wait_event.clear() self._client.connector.register( - MessageEndpoints.gui_instruction_response(request_id), - cb=self._on_rpc_response, - parent=self, + MessageEndpoints.gui_instruction_response(request_id), cb=self._on_rpc_response ) self._client.connector.set_and_publish(MessageEndpoints.gui_instructions(receiver), rpc_msg) @@ -276,11 +274,10 @@ class RPCBase: self._rpc_response = None return self._create_widget_from_msg_result(msg_result) - @staticmethod - def _on_rpc_response(msg_obj: MessageObject, parent: RPCBase) -> None: + def _on_rpc_response(self, msg_obj: MessageObject) -> None: msg = cast(messages.RequestResponseMessage, msg_obj.value) - parent._rpc_response = msg - parent._msg_wait_event.set() + self._rpc_response = msg + self._msg_wait_event.set() def _create_widget_from_msg_result(self, msg_result): if msg_result is None: