From d123d3bb8212e449613f050574eb2e1a572b8c4e Mon Sep 17 00:00:00 2001 From: Matthias Date: Thu, 7 Dec 2023 07:25:34 +0100 Subject: [PATCH] Completely mock Torch on macos we already skip tests - now we also need to skip it's imports. --- tests/freqai/conftest.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/tests/freqai/conftest.py b/tests/freqai/conftest.py index d2383c5b9..8a9425a32 100644 --- a/tests/freqai/conftest.py +++ b/tests/freqai/conftest.py @@ -22,7 +22,17 @@ def is_mac() -> bool: @pytest.fixture(autouse=True) def patch_torch_initlogs(mocker) -> None: - mocker.patch("torch._logging._init_logs") + + if is_mac(): + # Mock torch import completely + import sys + import types + + module_name = 'torch' + mocked_module = types.ModuleType(module_name) + sys.modules[module_name] = mocked_module + else: + mocker.patch("torch._logging._init_logs") @pytest.fixture(scope="function")