diff --git a/requirements-freqai.txt b/requirements-freqai.txt index 0532562da..31366efa7 100644 --- a/requirements-freqai.txt +++ b/requirements-freqai.txt @@ -5,7 +5,7 @@ # Required for freqai scikit-learn==1.4.1.post1 joblib==1.3.2 -catboost==1.2.2; 'arm' not in platform_machine and python_version < '3.12' +catboost==1.2.3; 'arm' not in platform_machine lightgbm==4.3.0 xgboost==2.0.3 tensorboard==2.16.2 diff --git a/tests/freqai/test_freqai_interface.py b/tests/freqai/test_freqai_interface.py index 58648d97f..e3c286fcd 100644 --- a/tests/freqai/test_freqai_interface.py +++ b/tests/freqai/test_freqai_interface.py @@ -27,7 +27,7 @@ def is_arm() -> bool: def can_run_model(model: str) -> None: is_pytorch_model = 'Reinforcement' in model or 'PyTorch' in model - if is_py12() and ("Catboost" in model or is_pytorch_model): + if is_py12() and is_pytorch_model: pytest.skip("Model not supported on python 3.12 yet.") if is_arm() and "Catboost" in model: