Type Hints for Lightning Core

pytorch_lightning/core/saving.py

@@ -34,7 +34,7 @@ def on_hpc_load(self, checkpoint: Dict[str, Any]) -> None:
         Hook to do whatever you need right before Slurm manager loads the model
         """
 
-
+        
 def load_hparams_from_tags_csv(tags_csv: str) -> Namespace:
     if not os.path.isfile(tags_csv):
         log.warning(f'Missing Tags: {tags_csv}.')