Ruff formatting fixes

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: Maxusmusti

src/instructlab/training/utils.py

@@ -785,11 +785,10 @@ def _get_state_dict_patched(model, unwrap=False):
         elif getattr(model.module.config, "_fp8_scales", None):
             # FP8 model (e.g. Ministral): re-quantize state dict before saving
             if accelerator.is_main_process:
+                # First Party
                 from instructlab.training.vlm_utils import requantize_fp8_state_dict
 
-                log_rank_0(
-                    "Re-quantizing FP8 parameters for checkpoint compatibility"
-                )
+                log_rank_0("Re-quantizing FP8 parameters for checkpoint compatibility")
                 model_state = model.module.state_dict()
                 model_state = requantize_fp8_state_dict(
                     model_state, model.module.config._fp8_scales

src/instructlab/training/vlm_utils.py

@@ -2,13 +2,13 @@
 import logging
 
 # Third Party
-import torch
 from torch import nn
 from transformers import AutoConfig, AutoModelForImageTextToText, PreTrainedModel
 from transformers.models.auto import (
     MODEL_FOR_CAUSAL_LM_MAPPING,
     MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPING,
 )
+import torch
 
 logger = logging.getLogger("instructlab.training")