From e3f01c5e72154b3bd20a98898dbbe8c02a565e49 Mon Sep 17 00:00:00 2001
From: Sunli <scott_s829@163.com>
Date: Wed, 13 Mar 2024 18:31:29 +0800
Subject: [PATCH] Update mod.rs

---
 python/src/trade/mod.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/src/trade/mod.rs b/python/src/trade/mod.rs
index 62b9fd750..982b7b263 100644
--- a/python/src/trade/mod.rs
+++ b/python/src/trade/mod.rs
@@ -25,6 +25,7 @@ pub(crate) fn register_types(parent: &PyModule) -> PyResult<()> {
     parent.add_class::<types::OrderChargeItem>()?;
     parent.add_class::<types::OrderChargeDetail>()?;
     parent.add_class::<types::OrderDetail>()?;
+    parent.add_class::<types::BalanceType>()?;
     parent.add_class::<types::EstimateMaxPurchaseQuantityResponse>()?;
 
     parent.add_class::<context::TradeContext>()?;