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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: b10-transfer
3
- Version: 0.3.10
3
+ Version: 0.3.12
4
4
  Summary: Distributed PyTorch file transfer for Baseten - Environment-aware, lock-free file transfer management
5
5
  License: MIT
6
6
  Keywords: pytorch,file-transfer,cache,machine-learning,inference
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
4
4
 
5
5
  [tool.poetry]
6
6
  name = "b10-transfer"
7
- version = "0.3.10"
7
+ version = "0.3.12"
8
8
  description = "Distributed PyTorch file transfer for Baseten - Environment-aware, lock-free file transfer management"
9
9
  authors = ["Shounak Ray <shounak.noreply@baseten.co>", "Fred Liu <fred.liu.noreply@baseten.co>"]
10
10
  maintainers = ["Fred Liu <fred.liu.noreply@baseten.co>", "Shounak Ray <shounak.noreply@baseten.co>"]
@@ -9,7 +9,7 @@ from .constants import OperationStatus
9
9
  from .logging_utils import get_b10_logger
10
10
 
11
11
  # Version
12
- __version__ = "0.3.10"
12
+ __version__ = "0.3.12"
13
13
 
14
14
  __all__ = [
15
15
  "CacheError",
@@ -34,6 +34,9 @@ class Config:
34
34
  """
35
35
  default_dir = f"/tmp/torchinductor_{get_current_username()}"
36
36
  chosen = os.getenv("TORCHINDUCTOR_CACHE_DIR", default_dir)
37
+ # Expand paths
38
+ if chosen.startswith("~"):
39
+ os.path.expanduser("~") + chosen[1:]
37
40
  return validate_path_security(
38
41
  chosen,
39
42
  self._allowed_torch_cache_prefixes(),
File without changes