refactor(checkpoint_consumer): remove unused import of torch to streamline code

erfanMhi · erfanMhi · commit 10032c23e937 · 2025-12-18T14:11:06.000Z
diff --git a/grail/infrastructure/checkpoint_consumer.py b/grail/infrastructure/checkpoint_consumer.py
@@ -599,7 +599,6 @@ async def _reconstruct_from_delta(
         Returns:
             Path to reconstructed checkpoint directory, or None on failure
         """
-        import torch
         from safetensors.torch import load_file, save_file
 
         try:
