Skip to content

GitLab

Unverified Commit b0dbbd7a authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

Fix mypy job failing (#5970)

parent 970ba355
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
torchvision/models/_api.py
View file @ b0dbbd7a
import importlib import importlib
import inspect import inspect
import sys import sys
from collections import OrderedDict
from dataclasses import dataclass, fields from dataclasses import dataclass, fields
from inspect import signature from inspect import signature
from typing import Any, Callable, Dict, cast from typing import Any, Callable, Dict, cast
...@@ -60,7 +59,7 @@ class WeightsEnum(StrEnum): ...@@ -60,7 +59,7 @@ class WeightsEnum(StrEnum):
) )
return obj return obj
def get_state_dict(self, progress: bool) -> OrderedDict: def get_state_dict(self, progress: bool) -> Dict[str, Any]:
return load_state_dict_from_url(self.url, progress=progress) return load_state_dict_from_url(self.url, progress=progress)
def __repr__(self) -> str: def __repr__(self) -> str:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment