Apply isort and black reformatting

Signed-off-by: ananthsub <[email protected]>

nemo/lightning/io/pl.py

@@ -164,7 +164,11 @@ def save_checkpoint(self, checkpoint: Dict[str, Any], path: _PATH, storage_optio
         validate_sharding_integrity = not (self.validated_consistency and self.assume_constant_structure)
         self.validated_consistency = True
 
-        consistency_check = _skip_optimizer_states_before_consistancy_check if validate_sharding_integrity else _noop_preprocess_common_before_consistancy_check
+        consistency_check = (
+            _skip_optimizer_states_before_consistancy_check
+            if validate_sharding_integrity
+            else _noop_preprocess_common_before_consistancy_check
+        )
 
         return dist_checkpointing.save(
             sharded_state_dict=checkpoint,
@@ -337,5 +341,6 @@ def _skip_optimizer_states_before_consistancy_check(state_dict: Dict[str, Any])
     sd.pop("optimizer_states", None)
     return sd
 
+
 def _noop_preprocess_common_before_consistancy_check(state_dict: Dict[str, Any]) -> Dict[str, Any]:
-    return state_dict
+    return state_dict