b10-transfer 0.3.11__tar.gz → 0.3.12__tar.gz

{b10_transfer-0.3.11 → b10_transfer-0.3.12}/PKG-INFO

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

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

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

{b10_transfer-0.3.11 → b10_transfer-0.3.12}/src/b10_transfer/cache.py

@@ -166,7 +166,6 @@ def load_compile_cache() -> OperationStatus:
     """
     with cache_operation("Load"):
         b10fs_dir, torch_dir, work_dir = _setup_cache_paths()
-        logger.info(f"TORCHINDUCTOR_CACHE_DIR IS {config.TORCH_CACHE_DIR}")
 
         cache_filename = get_cache_filename()
         final_file, _ = _get_cache_file_paths(cache_filename, b10fs_dir)

{b10_transfer-0.3.11 → b10_transfer-0.3.12}/src/b10_transfer/cache_cli.py

@@ -89,8 +89,6 @@ def main() -> None:
 
     logger = _setup_logging(cfg.loglevel)
 
-    logger.info(f"TORCHINDUCTOR_CACHE_DIR IS {config.TORCH_CACHE_DIR}")
-
     # 1) Preload any existing cache (non-fatal on error)
     try:
         if load_compile_cache() == OperationStatus.SUCCESS:
@@ -116,6 +114,5 @@ def main() -> None:
     except Exception as e:
         logger.exception("save_compile_cache() failed: %s", e)
         sys.exit(3)
-    logger.info(f"TORCHINDUCTOR_CACHE_DIR IS {config.TORCH_CACHE_DIR}")
 
     logger.info("vLLM automatic torch compile cache done.")

{b10_transfer-0.3.11 → b10_transfer-0.3.12}/src/b10_transfer/config.py

@@ -34,6 +34,9 @@ class Config:
         """
         default_dir = f"/tmp/torchinductor_{get_current_username()}"
         chosen = os.getenv("TORCHINDUCTOR_CACHE_DIR", default_dir)
+        # Expand paths
+        if chosen.startswith("~"):
+            os.path.expanduser("~") + chosen[1:]
         return validate_path_security(
             chosen,
             self._allowed_torch_cache_prefixes(),