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 = [ diff --git a/requirements-freqai-rl.txt b/requirements-freqai-rl.txt index 37655e657..b077300df 100644 --- a/requirements-freqai-rl.txt +++ b/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