Skip to content

Commit 43c70ec

Browse files
authored
Fix optimizers overloads typing annotation (#10069)
1 parent b376799 commit 43c70ec

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

pytorch_lightning/core/lightning.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,12 @@ def optimizers(self, use_pl_optimizer: Literal[True] = True) -> Union[LightningO
130130
def optimizers(self, use_pl_optimizer: Literal[False]) -> Union[Optimizer, List[Optimizer]]:
131131
...
132132

133+
@overload
134+
def optimizers(
135+
self, use_pl_optimizer: bool
136+
) -> Union[Optimizer, LightningOptimizer, List[Optimizer], List[LightningOptimizer]]:
137+
...
138+
133139
def optimizers(
134140
self, use_pl_optimizer: bool = True
135141
) -> Union[Optimizer, LightningOptimizer, List[Optimizer], List[LightningOptimizer]]:

0 commit comments

Comments
 (0)