Update community_import.py

lint

Signed-off-by: Stan Kirdey <stan@inflection.ai>

Author: skirdey-inflection

nemo_rl/models/megatron/community_import.py

@@ -105,21 +105,25 @@ def export_model_from_megatron(
         )
 
     try:
-        from megatron.bridge.training.model_load_save import temporary_distributed_context
+        from megatron.bridge.training.model_load_save import (
+            temporary_distributed_context,
+        )
     except ImportError:
         raise ImportError("megatron.bridge.training is not available.")
 
     bridge = AutoBridge.from_hf_pretrained(hf_model_name, trust_remote_code=True)
-    
+
     # Export performs on CPU with proper distributed context
     with temporary_distributed_context(backend="gloo"):
         # Load the Megatron model
-        megatron_model = bridge.load_megatron_model(input_path, skip_temp_dist_context=True)
-        
+        megatron_model = bridge.load_megatron_model(
+            input_path, skip_temp_dist_context=True
+        )
+
         # Save in HuggingFace format
         bridge.save_hf_pretrained(megatron_model, output_path)
 
     # resetting mcore state
-    import megatron.core.rerun_state_machine
+    import megatron.core.rerun_state_machinec
 
     megatron.core.rerun_state_machine.destroy_rerun_state_machine()