Merge pull request #12400 from qinhanlei/patch-1

Update sb3_contrib dependency condition

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 = [

requirements-freqai-rl.txt

@@ -6,6 +6,6 @@ torch==2.9.0; sys_platform != 'darwin' or platform_machine != 'x86_64'
 gymnasium==1.2.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