b10-transfer 0.3.8__tar.gz → 0.3.9__tar.gz

{b10_transfer-0.3.8 → b10_transfer-0.3.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: b10-transfer
-Version: 0.3.8
+Version: 0.3.9
 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.8 → b10_transfer-0.3.9}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "b10-transfer"
-version = "0.3.8"
+version = "0.3.9"
 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.8 → b10_transfer-0.3.9}/src/b10_transfer/__init__.py

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

{b10_transfer-0.3.8 → b10_transfer-0.3.9}/src/b10_transfer/utils.py

@@ -226,9 +226,9 @@ def critical_section_b10fs_file_lock(name):
     def decorator(func):
         def wrapper(*args, **kwargs):
             # Import here to avoid circular dependency
-            from .constants import B10FS_CACHE_DIR
+            from .config import config
 
-            lock_dir = Path(B10FS_CACHE_DIR)
+            lock_dir = Path(config.B10FS_CACHE_DIR)
             lock_dir.mkdir(parents=True, exist_ok=True)
 
             lock_file = lock_dir / f"{name}.lock"