b10-transfer 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

b10_transfer/__init__.py

@@ -1,27 +1,13 @@
-"""B10 Transfer - Lock-free PyTorch compilation cache for Baseten."""
+"""B10 Transfer - Lock-free PyTorch file transfer for Baseten."""
 
-from .core import transfer
-from .torch_cache import load_compile_cache, save_compile_cache, clear_local_cache
-from .async_transfers import (
-    start_transfer_async,
-    get_transfer_status,
-    is_transfer_complete,
-    wait_for_completion,
-    cancel_transfer,
-    list_active_transfers,
-    TransferProgress,
-)
-from .async_torch_cache import (
-    load_compile_cache_async,
-    save_compile_cache_async,
-)
+from .core import load_compile_cache, save_compile_cache, clear_local_cache
 from .utils import CacheError, CacheValidationError
 from .space_monitor import CacheOperationInterrupted
 from .info import get_cache_info, list_available_caches
-from .constants import SaveStatus, LoadStatus, TransferStatus, AsyncTransferStatus
+from .constants import SaveStatus, LoadStatus
 
 # Version
-__version__ = "0.1.1"
+__version__ = "0.1.3"
 
 __all__ = [
     "CacheError",
@@ -29,23 +15,9 @@ __all__ = [
     "CacheOperationInterrupted",
     "SaveStatus",
     "LoadStatus",
-    "TransferStatus",
-    "AsyncTransferStatus",
-    "transfer",
     "load_compile_cache",
     "save_compile_cache",
     "clear_local_cache",
     "get_cache_info",
     "list_available_caches",
-    # Generic async operations
-    "start_transfer_async",
-    "get_transfer_status",
-    "is_transfer_complete",
-    "wait_for_completion",
-    "cancel_transfer",
-    "list_active_transfers",
-    "TransferProgress",
-    # Torch-specific async operations
-    "load_compile_cache_async",
-    "save_compile_cache_async",
 ]

b10_transfer/constants.py

@@ -36,8 +36,7 @@ B10FS_CACHE_DIR = validate_path_security(
     _b10fs_cache_dir, [_REQUIRED_TORCH_CACHE_DIR_PREFIX], "B10FS_CACHE_DIR"
 )
 
-# Validate LOCAL_WORK_DIR - allow /app, /tmp, and /cache paths.
-# This is like a "scratch" directory where you can do work (like compression/archival for example)
+# Validate LOCAL_WORK_DIR - allow /app, /tmp, and /cache paths
 _local_work_dir = os.getenv("LOCAL_WORK_DIR", "/app")
 LOCAL_WORK_DIR = validate_path_security(
     _local_work_dir, ["/app/", "/tmp/", "/cache/"], "LOCAL_WORK_DIR"
@@ -113,7 +112,6 @@ class WorkerStatus(Enum):
     SUCCESS = auto()
     ERROR = auto()
     CANCELLED = auto()
-    FILE_NOT_FOUND = auto()
 
 
 class LoadStatus(Enum):
@@ -134,19 +132,7 @@ class SaveStatus(Enum):
 
 
 class TransferStatus(Enum):
-    """Status values for generic transfer operations."""
+    """Status values for file transfer operations."""
 
     SUCCESS = auto()
     ERROR = auto()
-    INTERRUPTED = auto()
-    DOES_NOT_EXIST = auto()
-
-
-class AsyncTransferStatus(Enum):
-    NOT_STARTED = auto()
-    IN_PROGRESS = auto()
-    SUCCESS = auto()
-    ERROR = auto()
-    INTERRUPTED = auto()
-    CANCELLED = auto()
-    DOES_NOT_EXIST = auto()