Skip to content

Commit 5edf4b1

Browse files
committed
fix type annotation
1 parent d1526f7 commit 5edf4b1

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/lightning/pytorch/loggers/utilities.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -149,11 +149,11 @@ def __init__(self, __iterable: Optional[Union[Mapping[str, _T], Iterable[_T]]] =
149149
super().__init__(__iterable.values())
150150
_dict = dict(zip(__iterable.keys(), range(len(__iterable))))
151151
else:
152-
default_dict = {}
152+
default_dict: dict[str, int] = {}
153153
if isinstance(__iterable, _ListMap):
154154
default_dict = __iterable._dict.copy()
155155
super().__init__(() if __iterable is None else __iterable)
156-
_dict: dict = default_dict
156+
_dict = default_dict
157157
self._dict = _dict
158158

159159
def __eq__(self, other: Any) -> bool:
@@ -183,7 +183,7 @@ def pop(self, key: Union[str, SupportsIndex], default: _T, /) -> _T: ...
183183
@overload
184184
def pop(self, key: str, default: _PT, /) -> Union[_T, _PT]: ...
185185

186-
def pop(self, key=-1, default=None):
186+
def pop(self, key: int = -1, default: Optional = None) -> _T:
187187
if isinstance(key, int):
188188
ret = super().pop(key)
189189
for str_key, idx in list(self._dict.items()):

0 commit comments

Comments
 (0)