Don't run Torch models on macos.

This commit is contained in:
Matthias
2024-03-12 07:30:06 +01:00
parent 4e02d31cdc
commit e560f9963c

View File

@@ -27,7 +27,7 @@ def can_run_model(model: str) -> None:
if is_arm() and "Catboost" in model:
pytest.skip("CatBoost is not supported on ARM.")
if is_pytorch_model and is_mac() and not is_arm():
if is_pytorch_model and is_mac():
pytest.skip("Reinforcement learning / PyTorch module not available on intel based Mac OS.")