adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

skip pytorch tests on python 3.11 and intel based mac os

Browse Source
This commit is contained in:
Yinon Polak 2023-03-21 14:34:27 +02:00
parent a80afc8f1b
commit 3fa23860c0
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/freqai/test_freqai_interface.py
View File

@ -34,13 +34,14 @@ 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() or is_py11()) and "Catboost" in model:
pytest.skip("CatBoost is not supported on ARM") pytest.skip("CatBoost is not supported on ARM.")
if is_mac() and not is_arm() and 'Reinforcement' in model: is_pytorch_model = 'Reinforcement' in model or 'PyTorch' in model
pytest.skip("Reinforcement learning module not available on intel based Mac OS") if is_pytorch_model and is_mac() and not is_arm():
pytest.skip("Reinforcement learning / PyTorch module not available on intel based Mac OS.")
if is_py11() and 'Reinforcement' in model: if is_pytorch_model and is_py11():
pytest.skip("Reinforcement learning currently not available on python 3.11.") pytest.skip("Reinforcement learning / PyTorch currently not available on python 3.11.")
@pytest.mark.parametrize('model, pca, dbscan, float32, can_short, shuffle, buffer', [ @pytest.mark.parametrize('model, pca, dbscan, float32, can_short, shuffle, buffer', [