Skip to content

Commit 6ebdcd1

Browse files
committed
typing
1 parent d03f217 commit 6ebdcd1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchft/optim.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
1313
"""
1414

15-
from typing import TYPE_CHECKING, Any, List, Mapping, Optional
15+
from typing import TYPE_CHECKING, Any, Dict, List, Optional
1616

1717
import torch
1818
from torch.optim import Optimizer
@@ -55,7 +55,7 @@ def step(self, closure: Optional[object] = None) -> None:
5555
self.optim.step()
5656

5757
@property
58-
def param_groups(self) -> List[Mapping[str, Any]]:
58+
def param_groups(self) -> List[Dict[str, Any]]:
5959
return self.optim.param_groups
6060

6161
@property

0 commit comments

Comments
 (0)