ignore return type

Author: matsumotosan

src/lightning/fabric/strategies/fsdp.py

@@ -795,7 +795,7 @@ def _optimizer_has_flat_params(optimizer: Optimizer) -> bool:
     )
 
 
-def _get_sharded_state_dict_context(module: Module) -> Generator[None, None, None]:
+def _get_sharded_state_dict_context(module: Module) -> Generator:
     from torch.distributed.fsdp import FullyShardedDataParallel as FSDP
     from torch.distributed.fsdp.api import ShardedOptimStateDictConfig, ShardedStateDictConfig, StateDictType
 
@@ -806,7 +806,7 @@ def _get_sharded_state_dict_context(module: Module) -> Generator[None, None, Non
         state_dict_type=StateDictType.SHARDED_STATE_DICT,
         state_dict_config=state_dict_config,
         optim_state_dict_config=optim_state_dict_config,
-    )
+    )  # type: ignore[return-value]
 
 
 def _get_full_state_dict_context(
@@ -823,7 +823,7 @@ def _get_full_state_dict_context(
         state_dict_type=StateDictType.FULL_STATE_DICT,
         state_dict_config=state_dict_config,
         optim_state_dict_config=optim_state_dict_config,
-    )
+    )  # type: ignore[return-value]
 
 
 def _is_sharded_checkpoint(path: Path) -> bool: