diff --git a/tests/freqai/conftest.py b/tests/freqai/conftest.py index 6f3c3018d..4a1976d9d 100644 --- a/tests/freqai/conftest.py +++ b/tests/freqai/conftest.py @@ -41,7 +41,7 @@ def patch_torch_initlogs(mocker) -> None: module_name = 'torch' mocked_module = types.ModuleType(module_name) sys.modules[module_name] = mocked_module - elif not is_py12(): + else: mocker.patch("torch._logging._init_logs")