fix: fmt + lint

Signed-off-by: Will Johnson <mwjohnson728@gmail.com>

Author: willmj

plugins/accelerated-moe/src/fms_acceleration_moe/utils/scattermoe_prepare.py

@@ -255,7 +255,7 @@ def prepare_scattermoe(
                 module_name,
                 router_name,
                 "|".join(expert_name),
-                lora_start=lora
+                lora_start=lora,
                 target_modules=lora_config.target_modules,
             )
 

plugins/accelerated-moe/src/fms_acceleration_moe/utils/scattermoe_state_dict.py

@@ -88,7 +88,7 @@ def get_checkpoint_meta_from_sharded_safetensor(
     router_name: str = "gate",  # e.g., named "gate" within block_sparse_moe
     expert_name: str = "experts",  # e.g., named "experts" within block_sparse_moe
     expert_map: Dict = None,  # map -> [w1,w2,w3]
-    lora_start: bool = False, # if lora is detected in prepare_scattermoe.py
+    lora_start: bool = False,  # if lora is detected in prepare_scattermoe.py
     lora_utils: bool = False,  # if lora is detected in checkpoint_utils.py
     target_modules: Dict = None,  # target modules from prepare_scattermoe.py
 ) -> Dict[str, List[Tuple]]:
@@ -197,10 +197,8 @@ def _insert(L: List, i: int, v):
             elif not lora_utils and not lora_start:
                 for mod in expert_map.get(m.group(1), expert_map.get(m.group(3))):
                     _insert(_map[f"{mod}.weight"], index, (k, stfile))
-            
-            assert mod is not None, f"cannot map '{rel_k}'"
 
-                
+            assert mod is not None, f"cannot map '{rel_k}'"
 
     if len(_map) == 0:
         raise ValueError(