Apply isort and black reformatting

Signed-off-by: ashors1 <[email protected]>
NVIDIA · Dec 2, 2024 · ee523a3 · ee523a3
1 parent 05d8bf2
commit ee523a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/lightning/io/pl.py b/nemo/lightning/io/pl.py
@@ -307,7 +307,7 @@ def drop_optimizer_states(self, path):
         from megatron.core import dist_checkpointing
 
         dist_checkpointing.remove_sharded_tensors(path, key_prefix="optimizer")
-    
+
     def adjust_non_strict_load(self, path: _PATH, sharded_state_dict: Dict[str, Any]):
         from megatron.core import dist_checkpointing
         from megatron.core.dist_checkpointing.dict_utils import extract_matching_values