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