Bump catboost to 1.2, disable some constraints

This commit is contained in:
Matthias 2023-05-02 07:23:38 +02:00
parent 0db1869356
commit e76356aff5
2 changed files with 2 additions and 2 deletions

View File

@ -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

View File

@ -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