Improve Polymarket account update on trade update

nautechsystems · Sep 27, 2024 · ac06b8a · ac06b8a
1 parent de332c3
commit ac06b8a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nautilus_trader/adapters/polymarket/execution.py b/nautilus_trader/adapters/polymarket/execution.py
@@ -915,6 +915,8 @@ def _handle_ws_trade_msg(self, msg: PolymarketUserTrade, wait_for_ack: bool):
             self._send_fill_report(report)
             return
 
+        self._loop.create_task(self._update_account_state())
+
         order = self._cache.order(client_order_id)
         if order.is_closed:
             return  # Already closed (only status update)
@@ -940,5 +942,3 @@ def _handle_ws_trade_msg(self, msg: PolymarketUserTrade, wait_for_ack: bool):
             ts_event=millis_to_nanos(int(msg.match_time)),
             info=msgspec.structs.asdict(msg),
         )
-
-        self._loop.create_task(self._update_account_state())