From: Jérôme Benoit <jerome.benoit@piment-noir.org>
Date: Tue, 25 Mar 2025 15:42:59 +0000 (+0100)
Subject: fix: handle margin trading mode
X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=4fbee9ba012e1b3024a59d0ec8c324324b736ec7;p=freqai-strategies.git

fix: handle margin trading mode

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
---

diff --git a/ReforceXY/user_data/strategies/RLAgentStrategy.py b/ReforceXY/user_data/strategies/RLAgentStrategy.py
index 5a81790..7049e1d 100644
--- a/ReforceXY/user_data/strategies/RLAgentStrategy.py
+++ b/ReforceXY/user_data/strategies/RLAgentStrategy.py
@@ -125,7 +125,7 @@ class RLAgentStrategy(IStrategy):
 
     def is_short_allowed(self) -> bool:
         trading_mode = self.config.get("trading_mode")
-        if trading_mode == "futures":
+        if trading_mode == "margin" or trading_mode == "futures":
             return True
         elif trading_mode == "spot":
             return False
diff --git a/quickadapter/user_data/strategies/QuickAdapterV3.py b/quickadapter/user_data/strategies/QuickAdapterV3.py
index c53ae58..0e2204c 100644
--- a/quickadapter/user_data/strategies/QuickAdapterV3.py
+++ b/quickadapter/user_data/strategies/QuickAdapterV3.py
@@ -645,7 +645,7 @@ class QuickAdapterV3(IStrategy):
 
     def is_short_allowed(self) -> bool:
         trading_mode = self.config.get("trading_mode")
-        if trading_mode == "futures":
+        if trading_mode == "margin" or trading_mode == "futures":
             return True
         elif trading_mode == "spot":
             return False