Bump catboost to 1.2, disable some constraints
This commit is contained in:
@@ -5,7 +5,7 @@
|
|||||||
# Required for freqai
|
# Required for freqai
|
||||||
scikit-learn==1.1.3
|
scikit-learn==1.1.3
|
||||||
joblib==1.2.0
|
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
|
lightgbm==3.3.5
|
||||||
xgboost==1.7.5
|
xgboost==1.7.5
|
||||||
tensorboard==2.13.0
|
tensorboard==2.13.0
|
||||||
|
|||||||
@@ -34,7 +34,7 @@ def is_mac() -> bool:
|
|||||||
|
|
||||||
|
|
||||||
def can_run_model(model: str) -> None:
|
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.")
|
pytest.skip("CatBoost is not supported on ARM.")
|
||||||
|
|
||||||
is_pytorch_model = 'Reinforcement' in model or 'PyTorch' in model
|
is_pytorch_model = 'Reinforcement' in model or 'PyTorch' in model
|
||||||
|
|||||||
Reference in New Issue
Block a user