b10-transfer 0.3.8__py3-none-any.whl → 0.3.10__py3-none-any.whl

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.
b10_transfer/__init__.py CHANGED
@@ -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.8"
12
+ __version__ = "0.3.10"
13
13
 
14
14
  __all__ = [
15
15
  "CacheError",
b10_transfer/cache_cli.py CHANGED
@@ -78,7 +78,7 @@ def _wait_for_ready(cfg: WaitCfg, logger: logging.Logger) -> bool:
78
78
 
79
79
  def main() -> None:
80
80
  vllm_cache_dir = os.getenv("VLLM_CACHE_ROOT", "~/.cache/vllm")
81
- os.environ["TORCHINDUCTOR_CACHE"] = vllm_cache_dir
81
+ os.environ["TORCHINDUCTOR_CACHE_DIR"] = vllm_cache_dir
82
82
 
83
83
  cfg = WaitCfg(
84
84
  url=DEFAULT_URL,
b10_transfer/utils.py CHANGED
@@ -226,9 +226,9 @@ def critical_section_b10fs_file_lock(name):
226
226
  def decorator(func):
227
227
  def wrapper(*args, **kwargs):
228
228
  # Import here to avoid circular dependency
229
- from .constants import B10FS_CACHE_DIR
229
+ from .config import config
230
230
 
231
- lock_dir = Path(B10FS_CACHE_DIR)
231
+ lock_dir = Path(config.B10FS_CACHE_DIR)
232
232
  lock_dir.mkdir(parents=True, exist_ok=True)
233
233
 
234
234
  lock_file = lock_dir / f"{name}.lock"
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: b10-transfer
3
- Version: 0.3.8
3
+ Version: 0.3.10
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
@@ -1,7 +1,7 @@
1
- b10_transfer/__init__.py,sha256=8cna_aMOPTQTLY3b_7pSFe9sSLft5FfqSCso7uRcpBA,729
1
+ b10_transfer/__init__.py,sha256=k3UP0x__MVeqzEkVc02nKifzQ57-MQ5sbBQZg2Fxgvs,730
2
2
  b10_transfer/archive.py,sha256=RGk7pmOdF24aATKygkFOAfHB-90arnW67nj-WURZfcw,6424
3
3
  b10_transfer/cache.py,sha256=VbAQx935rqdMXu6ejloa6Jw3n2KxkSMp3VLrNrgP--k,17480
4
- b10_transfer/cache_cli.py,sha256=29H7HPKOvHP2LVI6J1Gad42iTXAZIwVqjvrZyIuCb_Q,3378
4
+ b10_transfer/cache_cli.py,sha256=oSEAirpMEOpJ1PA80aisAaiU31kDpnCIhuo9pYXtLqA,3382
5
5
  b10_transfer/cleanup.py,sha256=2LXiNQH08xT7g5LtqAp7fiUYa1Rle_gM8LKxj5NbAwc,6337
6
6
  b10_transfer/config.py,sha256=z_7emRsb-IG7_KbGy9jLtzkbIkLGEbKpWDUPWC_PB58,4661
7
7
  b10_transfer/constants.py,sha256=oBfAvw2QyCRS1rFD9g1kDaToQqRX2bcnCVBUnCYD8uQ,1323
@@ -10,8 +10,8 @@ b10_transfer/environment.py,sha256=7DcEFmuxEMWwRM0M92q8al5rx8mpGY8dZ81Hokhb9B0,5
10
10
  b10_transfer/info.py,sha256=MR6gXvL3gBImnafuDgbOsZAHy9_akwZVlyIPGu2t8jQ,6345
11
11
  b10_transfer/logging_utils.py,sha256=c7iKNK9daNsrkHgsKl32rXscqqmWsq3a8ttz9M5ev3o,3460
12
12
  b10_transfer/space_monitor.py,sha256=24rClldo6EYaeikEla3_q4IwwI0r3DEPzaIbfU4hTRE,10746
13
- b10_transfer/utils.py,sha256=CRMnhswWdKfiZ6-TmLAvhMLFnVoygU9c25bjAqZv4T4,12028
14
- b10_transfer-0.3.8.dist-info/METADATA,sha256=2R3pMbKYIMNVkgXjv07XzvcpFy_Qk_grwUGjDFxwPnI,4108
15
- b10_transfer-0.3.8.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
16
- b10_transfer-0.3.8.dist-info/entry_points.txt,sha256=AFH8EfkeBv6ZuarnaUvbTjvySylf3GCSioYqB_ijrH8,65
17
- b10_transfer-0.3.8.dist-info/RECORD,,
13
+ b10_transfer/utils.py,sha256=cr-EEOtfu3xbyafneNFfiDoJrxCO16zXt6ZRHqSUd04,12023
14
+ b10_transfer-0.3.10.dist-info/METADATA,sha256=kQtKgKHfb_WhQy-jpMFnNO212G-yDHiUSmyTNxRr76Q,4109
15
+ b10_transfer-0.3.10.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
16
+ b10_transfer-0.3.10.dist-info/entry_points.txt,sha256=AFH8EfkeBv6ZuarnaUvbTjvySylf3GCSioYqB_ijrH8,65
17
+ b10_transfer-0.3.10.dist-info/RECORD,,