b10-transfer 0.3.11__py3-none-any.whl → 0.3.12__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.11"
+__version__ = "0.3.12"
 
 __all__ = [
     "CacheError",

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/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/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(),

{b10_transfer-0.3.11.dist-info → b10_transfer-0.3.12.dist-info}/METADATA

@@ -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.dist-info → b10_transfer-0.3.12.dist-info}/RECORD

@@ -1,9 +1,9 @@
-b10_transfer/__init__.py,sha256=BXQMvMXIeDQL7EWmjc5yT50EBPAOwMHmgLojrpOw_lc,730
+b10_transfer/__init__.py,sha256=ZABEbvVy5nDNNHPuY4oakpBfQR-VNtV4AIbJzIYCE_Y,730
 b10_transfer/archive.py,sha256=RGk7pmOdF24aATKygkFOAfHB-90arnW67nj-WURZfcw,6424
-b10_transfer/cache.py,sha256=LE6epqM0iAb7ls7Cle15U6VJwkhetuupJsT_CitJrxw,17556
-b10_transfer/cache_cli.py,sha256=0CK2x9K6E7nTewXmlLVK1GDVp-zZ2jzZzhql_ueejsc,3527
+b10_transfer/cache.py,sha256=VbAQx935rqdMXu6ejloa6Jw3n2KxkSMp3VLrNrgP--k,17480
+b10_transfer/cache_cli.py,sha256=oSEAirpMEOpJ1PA80aisAaiU31kDpnCIhuo9pYXtLqA,3382
 b10_transfer/cleanup.py,sha256=2LXiNQH08xT7g5LtqAp7fiUYa1Rle_gM8LKxj5NbAwc,6337
-b10_transfer/config.py,sha256=z_7emRsb-IG7_KbGy9jLtzkbIkLGEbKpWDUPWC_PB58,4661
+b10_transfer/config.py,sha256=DNmu5I2GWkiu03YHqEhCoTPrjp1DbU0Ifb2_oOk-h5I,4768
 b10_transfer/constants.py,sha256=oBfAvw2QyCRS1rFD9g1kDaToQqRX2bcnCVBUnCYD8uQ,1323
 b10_transfer/core.py,sha256=r79CI8Kpw9FyT17qYOze4tg4UyoL_EQXiucX5YiqXEM,4659
 b10_transfer/environment.py,sha256=7DcEFmuxEMWwRM0M92q8al5rx8mpGY8dZ81Hokhb9B0,5577
@@ -11,7 +11,7 @@ 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=cr-EEOtfu3xbyafneNFfiDoJrxCO16zXt6ZRHqSUd04,12023
-b10_transfer-0.3.11.dist-info/METADATA,sha256=_crxf7pCEoQNepkSzfr_UagqfquNjpxAtE9LUSjDwm4,4109
-b10_transfer-0.3.11.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-b10_transfer-0.3.11.dist-info/entry_points.txt,sha256=AFH8EfkeBv6ZuarnaUvbTjvySylf3GCSioYqB_ijrH8,65
-b10_transfer-0.3.11.dist-info/RECORD,,
+b10_transfer-0.3.12.dist-info/METADATA,sha256=bnZ2MRj1N0skrCF5CadUhUgKJbYGyJcr6wlnuYTUyTo,4109
+b10_transfer-0.3.12.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+b10_transfer-0.3.12.dist-info/entry_points.txt,sha256=AFH8EfkeBv6ZuarnaUvbTjvySylf3GCSioYqB_ijrH8,65
+b10_transfer-0.3.12.dist-info/RECORD,,