Merge pull request #12400 from qinhanlei/patch-1

Update sb3_contrib dependency condition
This commit is contained in:
Matthias
2025-10-20 20:20:55 +02:00
committed by GitHub
2 changed files with 5 additions and 5 deletions

View File

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

View File

@@ -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