From 5d227ff4809ac6e97a1a959fcb8c1e7d853c217b Mon Sep 17 00:00:00 2001 From: Luca Antiga Date: Mon, 25 Nov 2024 11:35:10 +0100 Subject: [PATCH] Update src/lightning/fabric/strategies/strategy.py --- src/lightning/fabric/strategies/strategy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lightning/fabric/strategies/strategy.py b/src/lightning/fabric/strategies/strategy.py index 3b3801a3e8..1788ffc757 100644 --- a/src/lightning/fabric/strategies/strategy.py +++ b/src/lightning/fabric/strategies/strategy.py @@ -147,7 +147,7 @@ class Strategy(ABC): def setup_module_and_optimizers( self, module: Module, optimizers: list[Optimizer], scheduler: Optional[_LRScheduler] = None - ) -> Tuple[Module, list[Optimizer], Optional[_LRScheduler]]: + ) -> tuple[Module, list[Optimizer], Optional[_LRScheduler]]: """Set up a model and multiple optimizers together. The returned objects are expected to be in the same order they were passed in. The default implementation will