fix ruff errors

Author: mtake

src/instructlab/training/gpt_oss_utils_correct.py

@@ -411,7 +411,9 @@ def is_gpt_oss(model_path_or_config: str | PretrainedConfig) -> bool:
     return getattr(model_config, "model_type", None) == "gpt_oss"
 
 
-def is_known_model(model_path_or_config: str | PretrainedConfig, known_model_type: str | list[str]) -> bool:
+def is_known_model(
+    model_path_or_config: str | PretrainedConfig, known_model_type: str | list[str]
+) -> bool:
     """
     Determine if the model is a known model.
     """
@@ -425,7 +427,9 @@ def is_known_model(model_path_or_config: str | PretrainedConfig, known_model_typ
     if isinstance(model_path_or_config, str):
         model_config = AutoConfig.from_pretrained(model_path_or_config)
 
-    known_model_types = [known_model_type] if isinstance(known_model_type, str) else known_model_type
+    known_model_types = (
+        [known_model_type] if isinstance(known_model_type, str) else known_model_type
+    )
     return getattr(model_config, "model_type", None) in known_model_types
 
 

src/instructlab/training/main_ds.py

@@ -346,7 +346,7 @@ def main(args):
     #       GPT-OSS specifically
     # We don't want to use use_orig_params for GPT-OSS models
     fsdp_should_use_orig_params = False
-    if m.is_gpt_oss or m.is_granitemoehybrid:  # NOTE is this guard needed?
+    if m.is_gpt_oss or m.is_granitemoehybrid:
         frozen_router_params = freeze_router_params(m)
         if frozen_router_params:
             logger.info("🎯 Detected an MoE model - frozen router parameters")

src/instructlab/training/model.py

@@ -419,7 +419,7 @@ def compute_loss(
 
         # add the MoE auxiliary loss (currently we only support this for GPT-OSS)
         if (
-            (self.is_gpt_oss or self.is_granitemoehybrid)  # NOTE is this guard needed?
+            (self.is_gpt_oss or self.is_granitemoehybrid)
             and hasattr(output, "aux_loss")
             and output.aux_loss is not None
         ):