b10-transfer 0.3.6__tar.gz → 0.3.7__tar.gz

{b10_transfer-0.3.6 → b10_transfer-0.3.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: b10-transfer
-Version: 0.3.6
+Version: 0.3.7
 Summary: Distributed PyTorch file transfer for Baseten - Environment-aware, lock-free file transfer management
 License: MIT
 Keywords: pytorch,file-transfer,cache,machine-learning,inference

{b10_transfer-0.3.6 → b10_transfer-0.3.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "b10-transfer"
-version = "0.3.6"
+version = "0.3.7"
 description = "Distributed PyTorch file transfer for Baseten - Environment-aware, lock-free file transfer management"
 authors = ["Shounak Ray <shounak.noreply@baseten.co>", "Fred Liu <fred.liu.noreply@baseten.co>"]
 maintainers = ["Fred Liu <fred.liu.noreply@baseten.co>", "Shounak Ray <shounak.noreply@baseten.co>"]

{b10_transfer-0.3.6 → b10_transfer-0.3.7}/src/b10_transfer/__init__.py

@@ -9,7 +9,7 @@ from .constants import OperationStatus
 from .logging_utils import get_b10_logger
 
 # Version
-__version__ = "0.3.6"
+__version__ = "0.3.7"
 
 __all__ = [
     "CacheError",

{b10_transfer-0.3.6 → b10_transfer-0.3.7}/src/b10_transfer/cleanup.py

@@ -13,8 +13,6 @@ from typing import List
 from .config import config
 from .constants import (
     CACHE_INCOMPLETE_SUFFIX,
-    CLEANUP_LOCK_TIMEOUT_SECONDS,
-    CLEANUP_INCOMPLETE_TIMEOUT_SECONDS,
 )
 from .utils import safe_execute, safe_unlink
 from .logging_utils import get_b10_logger
@@ -108,8 +106,8 @@ def cooperative_cleanup_b10fs() -> None:
     removing files that are likely orphaned due to pod crashes or failures.
 
     Removes:
-    - Lock files older than CLEANUP_LOCK_TIMEOUT_SECONDS (*.lock)
-    - Incomplete cache files older than CLEANUP_INCOMPLETE_TIMEOUT_SECONDS (*.incomplete*)
+    - Lock files older than config.CLEANUP_LOCK_TIMEOUT_SECONDS (*.lock)
+    - Incomplete cache files older than config.CLEANUP_INCOMPLETE_TIMEOUT_SECONDS (*.incomplete*)
 
     Does NOT remove:
     - Final cache files (*.latest.tar.gz) - these are the actual cached results
@@ -124,13 +122,15 @@ def cooperative_cleanup_b10fs() -> None:
         return
 
     # Find and clean stale lock files
-    stale_locks = _find_stale_files(b10fs_dir, "*.lock", CLEANUP_LOCK_TIMEOUT_SECONDS)
+    stale_locks = _find_stale_files(
+        b10fs_dir, "*.lock", config.CLEANUP_LOCK_TIMEOUT_SECONDS
+    )
     cleaned_locks = _cleanup_files(stale_locks, "lock")
 
     # Find and clean stale incomplete cache files
     incomplete_pattern = f"*{CACHE_INCOMPLETE_SUFFIX}*"
     stale_incomplete = _find_stale_files(
-        b10fs_dir, incomplete_pattern, CLEANUP_INCOMPLETE_TIMEOUT_SECONDS
+        b10fs_dir, incomplete_pattern, config.CLEANUP_INCOMPLETE_TIMEOUT_SECONDS
     )
     cleaned_incomplete = _cleanup_files(stale_incomplete, "incomplete cache")