diff --git a/requirements-freqai-rl.txt b/requirements-freqai-rl.txt index fba25d409..6256c387a 100644 --- a/requirements-freqai-rl.txt +++ b/requirements-freqai-rl.txt @@ -2,7 +2,7 @@ -r requirements-freqai.txt # Required for freqai-rl -torch==2.0.1 +torch==2.1.1 #until these branches will be released we can use this gymnasium==0.29.1 stable_baselines3==2.2.1 diff --git a/tests/freqai/conftest.py b/tests/freqai/conftest.py index be208408f..8a9425a32 100644 --- a/tests/freqai/conftest.py +++ b/tests/freqai/conftest.py @@ -20,6 +20,21 @@ def is_mac() -> bool: return "Darwin" in machine +@pytest.fixture(autouse=True) +def patch_torch_initlogs(mocker) -> None: + + 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") def freqai_conf(default_conf, tmp_path): freqaiconf = deepcopy(default_conf)