frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove allow-untyped-defs from onnx/_internal/_lazy_import.py (#143943)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143943
Approved by: https://github.com/justinchuby
This commit is contained in:
bobrenjc93
2024-12-28 13:39:25 -08:00
committed by PyTorch MergeBot
parent cf0b72c4ab
commit 7101b8ca35
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
torch/onnx/_internal/_lazy_import.py
View File

@ -1,6 +1,5 @@
"""Utility to lazily import modules."""
# mypy: allow-untyped-defs
from __future__ import annotations
import importlib
@ -17,7 +16,7 @@ class _LazyModule:
def __repr__(self) -> str:
return f"<lazy module '{self._name}'>"
def __getattr__(self, attr):
def __getattr__(self, attr: str) -> object:
if self._module is None:
self._module = importlib.import_module(".", self._name)
return getattr(self._module, attr)