b10-transfer 0.3.6__py3-none-any.whl → 0.3.8__py3-none-any.whl

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.8"
 
 __all__ = [
     "CacheError",

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")
 

b10_transfer/utils.py

@@ -305,9 +305,9 @@ def _is_b10fs_enabled() -> bool:
         bool: True if BASETEN_FS_ENABLED is set to "1" or "True", False otherwise.
     """
     # Import here to avoid circular dependency
-    from .constants import BASETEN_FS_ENABLED
+    from .config import config
 
-    return BASETEN_FS_ENABLED in ("1", "True", "true")
+    return config.BASETEN_FS_ENABLED in ("1", "True", "true")
 
 
 def _validate_b10fs_available() -> None:

{b10_transfer-0.3.6.dist-info → b10_transfer-0.3.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: b10-transfer
-Version: 0.3.6
+Version: 0.3.8
 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.dist-info → b10_transfer-0.3.8.dist-info}/RECORD

@@ -1,8 +1,8 @@
-b10_transfer/__init__.py,sha256=XQLvRmgyIdUxcijm3rKUESXOp6D7z_GE_m6y_tsBtbI,729
+b10_transfer/__init__.py,sha256=8cna_aMOPTQTLY3b_7pSFe9sSLft5FfqSCso7uRcpBA,729
 b10_transfer/archive.py,sha256=RGk7pmOdF24aATKygkFOAfHB-90arnW67nj-WURZfcw,6424
 b10_transfer/cache.py,sha256=VbAQx935rqdMXu6ejloa6Jw3n2KxkSMp3VLrNrgP--k,17480
 b10_transfer/cache_cli.py,sha256=29H7HPKOvHP2LVI6J1Gad42iTXAZIwVqjvrZyIuCb_Q,3378
-b10_transfer/cleanup.py,sha256=_i8WPwtnqiSD2CSLJfGOttE7SMJd_TUTnIqz4wI4pRE,6369
+b10_transfer/cleanup.py,sha256=2LXiNQH08xT7g5LtqAp7fiUYa1Rle_gM8LKxj5NbAwc,6337
 b10_transfer/config.py,sha256=z_7emRsb-IG7_KbGy9jLtzkbIkLGEbKpWDUPWC_PB58,4661
 b10_transfer/constants.py,sha256=oBfAvw2QyCRS1rFD9g1kDaToQqRX2bcnCVBUnCYD8uQ,1323
 b10_transfer/core.py,sha256=r79CI8Kpw9FyT17qYOze4tg4UyoL_EQXiucX5YiqXEM,4659
@@ -10,8 +10,8 @@ b10_transfer/environment.py,sha256=7DcEFmuxEMWwRM0M92q8al5rx8mpGY8dZ81Hokhb9B0,5
 b10_transfer/info.py,sha256=MR6gXvL3gBImnafuDgbOsZAHy9_akwZVlyIPGu2t8jQ,6345
 b10_transfer/logging_utils.py,sha256=c7iKNK9daNsrkHgsKl32rXscqqmWsq3a8ttz9M5ev3o,3460
 b10_transfer/space_monitor.py,sha256=24rClldo6EYaeikEla3_q4IwwI0r3DEPzaIbfU4hTRE,10746
-b10_transfer/utils.py,sha256=eyCckvl11xFOdsac4pjLEAleoahhIlBqAr81mwoGO-o,12036
-b10_transfer-0.3.6.dist-info/METADATA,sha256=shnBT3YJ29yof-COiJCr7dxPZMnOfekWgXnRuy0L9uw,4108
-b10_transfer-0.3.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-b10_transfer-0.3.6.dist-info/entry_points.txt,sha256=AFH8EfkeBv6ZuarnaUvbTjvySylf3GCSioYqB_ijrH8,65
-b10_transfer-0.3.6.dist-info/RECORD,,
+b10_transfer/utils.py,sha256=CRMnhswWdKfiZ6-TmLAvhMLFnVoygU9c25bjAqZv4T4,12028
+b10_transfer-0.3.8.dist-info/METADATA,sha256=2R3pMbKYIMNVkgXjv07XzvcpFy_Qk_grwUGjDFxwPnI,4108
+b10_transfer-0.3.8.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+b10_transfer-0.3.8.dist-info/entry_points.txt,sha256=AFH8EfkeBv6ZuarnaUvbTjvySylf3GCSioYqB_ijrH8,65
+b10_transfer-0.3.8.dist-info/RECORD,,