Avoid fully patching torch on M1 macs

This commit is contained in:
Matthias
2024-03-09 17:55:22 +01:00
parent 29f90cbd04
commit cc3b2000eb
2 changed files with 8 additions and 9 deletions

View File

@@ -25,10 +25,15 @@ def is_mac() -> bool:
return "Darwin" in machine
def is_arm() -> bool:
machine = platform.machine()
return "arm" in machine or "aarch64" in machine
@pytest.fixture(autouse=True)
def patch_torch_initlogs(mocker) -> None:
if is_mac():
if is_mac() and not is_arm():
# Mock torch import completely
import sys
import types

View File

@@ -1,5 +1,4 @@
import logging
import platform
import shutil
from pathlib import Path
from unittest.mock import MagicMock
@@ -15,13 +14,8 @@ from freqtrade.optimize.backtesting import Backtesting
from freqtrade.persistence import Trade
from freqtrade.plugins.pairlistmanager import PairListManager
from tests.conftest import EXMS, create_mock_trades, get_patched_exchange, log_has_re
from tests.freqai.conftest import (get_patched_freqai_strategy, is_mac, is_py12, make_rl_config,
mock_pytorch_mlp_model_training_parameters)
def is_arm() -> bool:
machine = platform.machine()
return "arm" in machine or "aarch64" in machine
from tests.freqai.conftest import (get_patched_freqai_strategy, is_arm, is_mac, is_py12,
make_rl_config, mock_pytorch_mlp_model_training_parameters)
def can_run_model(model: str) -> None: