Allow running catboost tests on 3.12

This commit is contained in:
Matthias 2024-03-09 18:16:22 +01:00
parent 971a81e15d
commit c5f2a69d9c

View File

@ -27,9 +27,6 @@ def is_arm() -> bool:
def can_run_model(model: str) -> None: def can_run_model(model: str) -> None:
is_pytorch_model = 'Reinforcement' in model or 'PyTorch' in model is_pytorch_model = 'Reinforcement' in model or 'PyTorch' in model
if is_py12() and ("Catboost" in model or is_pytorch_model):
pytest.skip("Model not supported on python 3.12 yet.")
if is_arm() 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.")