Skip to content

Commit faeda91

Browse files
carmoccaninginthecloud
authored andcommitted
Fix optimizers overloads typing annotation (Lightning-AI#10069)
1 parent 24db14e commit faeda91

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)