fix mypy (#7685)

58842057 · Philip Meier · GitHub · 016804ee · 58842057
Unverified Commit 58842057 authored Jun 21, 2023 by Philip Meier Committed by GitHub Jun 21, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

torchvision/models/_api.py torchvision/models/_api.py +3 -1

No files found.
--- a/torchvision/models/_api.py
+++ b/torchvision/models/_api.py
@@ -122,7 +122,9 @@ def get_weight(name: str) -> WeightsEnum:
    base_module_name = ".".join(sys.modules[__name__].__name__.split(".")[:-1])
    base_module = importlib.import_module(base_module_name)
    model_modules = [base_module] + [
-        x[1] for x in inspect.getmembers(base_module, inspect.ismodule) if x[1].__file__.endswith("__init__.py")
+        x[1]
+        for x in inspect.getmembers(base_module, inspect.ismodule)
+        if x[1].__file__.endswith("__init__.py")  # type: ignore[union-attr]
    ]

    weights_enum = None