Patch torch logging on 3.12

This commit is contained in:
Matthias 2024-03-12 19:59:46 +01:00
parent e560f9963c
commit 28bdbbffca

View File

@ -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")