From e76356aff5a55749a1e53f958804a9f0f9dc186b Mon Sep 17 00:00:00 2001 From: Matthias Date: Tue, 2 May 2023 07:23:38 +0200 Subject: [PATCH] Bump catboost to 1.2, disable some constraints --- requirements-freqai.txt | 2 +- tests/freqai/test_freqai_interface.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements-freqai.txt b/requirements-freqai.txt index e5bc23d56..9f6390e56 100644 --- a/requirements-freqai.txt +++ b/requirements-freqai.txt @@ -5,7 +5,7 @@ # Required for freqai scikit-learn==1.1.3 joblib==1.2.0 -catboost==1.1.1; platform_machine != 'aarch64' and 'arm' not in platform_machine and python_version < '3.11' +catboost==1.2; 'arm' not in platform_machine lightgbm==3.3.5 xgboost==1.7.5 tensorboard==2.13.0 diff --git a/tests/freqai/test_freqai_interface.py b/tests/freqai/test_freqai_interface.py index e27c8d2c0..95efaac52 100644 --- a/tests/freqai/test_freqai_interface.py +++ b/tests/freqai/test_freqai_interface.py @@ -34,7 +34,7 @@ def is_mac() -> bool: def can_run_model(model: str) -> None: - if (is_arm() or is_py11()) and "Catboost" in model: + if is_arm() and "Catboost" in model: pytest.skip("CatBoost is not supported on ARM.") is_pytorch_model = 'Reinforcement' in model or 'PyTorch' in model