chore: resolve ci linting issues.

Author: erfanMhi

grail/infrastructure/checkpoint_consumer.py

@@ -510,7 +510,6 @@ def _write_cached_checkpoint(
         metadata: CheckpointMetadata,
     ) -> None:
         """Write cached checkpoint synchronously (called from thread)."""
-        import torch
         from safetensors.torch import save_file
 
         tmp_dir = output_dir.parent / f"{output_dir.name}.partial"
@@ -1016,7 +1015,6 @@ async def _apply_delta_chain(
             Path to reconstructed checkpoint directory, or None on failure
         """
         import torch
-        from safetensors.torch import save_file
 
         try:
             # Load anchor weights

grail/neurons/miner.py

@@ -247,7 +247,7 @@ async def _run() -> None:
                         timers,
                         monitor,
                         self.use_drand,
-                        checkpoint_window,
+                        current_checkpoint_window,
                     )
 
                     if inferences:

grail/trainer/checkpoint_publisher.py

@@ -277,26 +277,19 @@ async def _compute_keep_windows(
 
     # Fetch anchor_window for each retained delta (the FULL checkpoint it depends on)
     anchor_windows = await asyncio.gather(
-        *(
-            _fetch_delta_anchor_window(delta_window, credentials)
-            for delta_window in deltas_to_keep
-        )
+        *(_fetch_delta_anchor_window(delta_window, credentials) for delta_window in deltas_to_keep)
     )
     for delta_window, anchor_window in zip(deltas_to_keep, anchor_windows, strict=False):
         if anchor_window is not None:
             keep.add(int(anchor_window))
-            logger.debug(
-                "Keeping anchor %d (required by delta %d)", anchor_window, delta_window
-            )
+            logger.debug("Keeping anchor %d (required by delta %d)", anchor_window, delta_window)
             continue
 
         # Fallback: keep nearest FULL checkpoint before this delta
         for full_window in full_windows:
             if full_window < delta_window:
                 keep.add(full_window)
-                logger.debug(
-                    "Keeping fallback anchor %d for delta %d", full_window, delta_window
-                )
+                logger.debug("Keeping fallback anchor %d for delta %d", full_window, delta_window)
                 break
 
     logger.info(

tests/unit/infrastructure/test_chained_deltas.py

@@ -9,9 +9,8 @@
 
 from __future__ import annotations
 
-from unittest.mock import AsyncMock, MagicMock, patch
+from unittest.mock import MagicMock, patch
 
-import pytest
 import torch
 
 from grail.infrastructure.checkpoint_consumer import CheckpointMetadata
@@ -21,7 +20,6 @@
     compute_weights_hash,
 )
 
-
 # ============================================================================
 # Tests for Chained Delta Precision (bit-exact bf16 reconstruction)
 # ============================================================================