Skip to content

Commit a7a5835

Browse files
authored
Update strategy.py
1 parent 9b45b99 commit a7a5835

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/fabric/strategies/strategy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def module_init_context(self, empty_init: Optional[bool] = None) -> ContextManag
146146

147147
def setup_module_and_optimizers(
148148
self, module: Module, optimizers: List[Optimizer], scheduler: Optional[_LRScheduler] = None
149-
) -> Tuple[Module, List[Optimizer]]:
149+
) -> Tuple[Module, List[Optimizer], Optional[_LRScheduler]]:
150150
"""Set up a model and multiple optimizers together.
151151
152152
The returned objects are expected to be in the same order they were passed in. The default implementation will

0 commit comments

Comments
 (0)