Patch torch logging on 3.12
This commit is contained in:
@@ -41,7 +41,7 @@ def patch_torch_initlogs(mocker) -> None:
|
|||||||
module_name = 'torch'
|
module_name = 'torch'
|
||||||
mocked_module = types.ModuleType(module_name)
|
mocked_module = types.ModuleType(module_name)
|
||||||
sys.modules[module_name] = mocked_module
|
sys.modules[module_name] = mocked_module
|
||||||
elif not is_py12():
|
else:
|
||||||
mocker.patch("torch._logging._init_logs")
|
mocker.patch("torch._logging._init_logs")
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user