b10-transfer 0.2.0__tar.gz → 0.2.2__tar.gz

{b10_transfer-0.2.0 → b10_transfer-0.2.2}/PKG-INFO

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

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

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

{b10_transfer-0.2.0 → b10_transfer-0.2.2}/src/b10_transfer/cache_cli.py

@@ -82,7 +82,7 @@ def main() -> None:
     os.environ["TORCHINDUCTOR_CACHE_DIR"] = VLLM_CACHE_DIR
 
     # Import here to allow environment variables to be set before the imported script uses them
-    from cache import load_compile_cache, save_compile_cache
+    from .cache import load_compile_cache, save_compile_cache
 
     cfg = WaitCfg(
         url=DEFAULT_URL,

{b10_transfer-0.2.0 → b10_transfer-0.2.2}/src/b10_transfer/constants.py

@@ -24,7 +24,7 @@ _torch_cache_dir = os.getenv(
     "TORCHINDUCTOR_CACHE_DIR", f"/tmp/torchinductor_{get_current_username()}"
 )
 TORCH_CACHE_DIR = validate_path_security(
-    _torch_cache_dir, ["/tmp/", "/cache/"], "TORCHINDUCTOR_CACHE_DIR"
+    _torch_cache_dir, ["/tmp/", "/cache/", "~/.cache"], "TORCHINDUCTOR_CACHE_DIR"
 )
 
 # B10FS cache directory validation