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 = [ freqai = [
"scikit-learn", "scikit-learn",
"joblib", "joblib",
'catboost; platform_machine != "aarch64"', "catboost; 'arm' not in platform_machine",
"lightgbm", "lightgbm",
"xgboost", "xgboost",
"tensorboard", "tensorboard",
"datasieve>=0.1.5", "datasieve>=0.1.5",
] ]
freqai_rl = [ freqai_rl = [
"torch", "torch; sys_platform != 'darwin' or platform_machine != 'x86_64'",
"gymnasium", "gymnasium",
"stable-baselines3", "stable-baselines3; sys_platform != 'darwin' or platform_machine != 'x86_64'",
"sb3-contrib", "sb3-contrib; sys_platform != 'darwin' or platform_machine != 'x86_64'",
"tqdm", "tqdm",
] ]
develop = [ develop = [

View File

@@ -6,6 +6,6 @@ torch==2.9.0; sys_platform != 'darwin' or platform_machine != 'x86_64'
gymnasium==1.2.1 gymnasium==1.2.1
# SB3 >=2.5.0 depends on torch 2.3.0 - which implies it dropped support x86 macos # 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' 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 # Progress bar for stable-baselines3 and sb3-contrib
tqdm==4.67.1 tqdm==4.67.1