From d6541d81c310b98c7d62324ff4260964db850216 Mon Sep 17 00:00:00 2001
From: WingStone <qinhanlei@gmail.com>
Date: Mon, 20 Oct 2025 10:49:35 +0800
Subject: [PATCH 1/2] Update sb3_contrib dependency condition

Install dependencies for development on macOS with Intel x86_64 Chip.
---
 requirements-freqai-rl.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements-freqai-rl.txt b/requirements-freqai-rl.txt
index d746f19ca..c0a271546 100644
--- a/requirements-freqai-rl.txt
+++ b/requirements-freqai-rl.txt
@@ -6,6 +6,6 @@ torch==2.8.0; sys_platform != 'darwin' or platform_machine != 'x86_64'
 gymnasium==0.29.1
 # SB3 >=2.5.0 depends on torch 2.3.0 - which implies it dropped support x86 macos
 stable_baselines3==2.7.0; sys_platform != 'darwin' or platform_machine != 'x86_64'
-sb3_contrib>=2.2.1
+sb3_contrib>=2.2.1; sys_platform != 'darwin' or platform_machine != 'x86_64'
 # Progress bar for stable-baselines3 and sb3-contrib
 tqdm==4.67.1

From 40bfaa624523e96ac3587fa0402ce61553eb0d26 Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Mon, 20 Oct 2025 19:41:50 +0200
Subject: [PATCH 2/2] chore: align pyproject.toml platform limitations

---
 pyproject.toml | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 4bac06296..6bf7fe362 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -85,17 +85,17 @@ hyperopt = [
 freqai = [
   "scikit-learn",
   "joblib",
-  'catboost; platform_machine != "aarch64"',
+  "catboost; 'arm' not in platform_machine",
   "lightgbm",
   "xgboost",
   "tensorboard",
   "datasieve>=0.1.5",
 ]
 freqai_rl = [
-  "torch",
+  "torch; sys_platform != 'darwin' or platform_machine != 'x86_64'",
   "gymnasium",
-  "stable-baselines3",
-  "sb3-contrib",
+  "stable-baselines3; sys_platform != 'darwin' or platform_machine != 'x86_64'",
+  "sb3-contrib; sys_platform != 'darwin' or platform_machine != 'x86_64'",
   "tqdm",
 ]
 develop = [