From 7fac782e3c74714b859b82a68b66b221f0218e54 Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Mon, 9 May 2022 08:42:15 +0200 Subject: [PATCH] update annotation --- torchvision/models/_api.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/torchvision/models/_api.py b/torchvision/models/_api.py index e47eaf73aab..05b35fe87f0 100644 --- a/torchvision/models/_api.py +++ b/torchvision/models/_api.py @@ -1,7 +1,6 @@ import importlib import inspect import sys -from collections import OrderedDict from dataclasses import dataclass, fields from inspect import signature from typing import Any, Callable, Dict, cast @@ -60,7 +59,7 @@ def verify(cls, obj: Any) -> Any: ) 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) def __repr__(self) -> str: