shared-tensor 0.2.15__tar.gz → 0.2.16__tar.gz

{shared_tensor-0.2.15 → shared_tensor-0.2.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: shared-tensor
-Version: 0.2.15
+Version: 0.2.16
 Summary: Native PyTorch CUDA IPC over Unix Domain Socket for same-host process separation
 Author-email: Athena Team <contact@world-sim-dev.org>
 Maintainer-email: Athena Team <contact@world-sim-dev.org>
@@ -403,6 +403,15 @@ info = provider.get_runtime_info()
 # execution_mode, server_socket_path, server_running, server_ready, server_info...
 ```
 
+## Logging
+
+`shared_tensor` now installs a default package logger on import and enables detailed logs by default.
+
+- default level: `INFO`
+- logger name: `shared_tensor`
+- override level with `SHARED_TENSOR_LOG_LEVEL`, for example `INFO`, `WARNING`, or `ERROR`
+- pass `verbose_debug=False` to `SharedTensorProvider`, `SharedTensorClient`, `AsyncSharedTensorClient`, or `SharedTensorServer` if you want to suppress detailed request-level logs
+
 ## Client Retry And Timeout Defaults
 
 The client now retries initial connection setup for up to `60s` when the server socket is not ready yet, covering the common server-startup race where the client starts slightly earlier.

{shared_tensor-0.2.15 → shared_tensor-0.2.16}/README.md

@@ -351,6 +351,15 @@ info = provider.get_runtime_info()
 # execution_mode, server_socket_path, server_running, server_ready, server_info...
 ```
 
+## Logging
+
+`shared_tensor` now installs a default package logger on import and enables detailed logs by default.
+
+- default level: `INFO`
+- logger name: `shared_tensor`
+- override level with `SHARED_TENSOR_LOG_LEVEL`, for example `INFO`, `WARNING`, or `ERROR`
+- pass `verbose_debug=False` to `SharedTensorProvider`, `SharedTensorClient`, `AsyncSharedTensorClient`, or `SharedTensorServer` if you want to suppress detailed request-level logs
+
 ## Client Retry And Timeout Defaults
 
 The client now retries initial connection setup for up to `60s` when the server socket is not ready yet, covering the common server-startup race where the client starts slightly earlier.

{shared_tensor-0.2.15 → shared_tensor-0.2.16}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "shared-tensor"
-version = "0.2.15"
+version = "0.2.16"
 description = "Native PyTorch CUDA IPC over Unix Domain Socket for same-host process separation"
 readme = "README.md"
 license = "MIT"

shared_tensor-0.2.16/shared_tensor/__init__.py

@@ -0,0 +1,51 @@
+"""shared_tensor: same-host same-GPU PyTorch CUDA IPC over local UDS RPC."""
+
+import logging
+import os
+
+
+class _SafeStreamHandler(logging.StreamHandler):
+    def emit(self, record: logging.LogRecord) -> None:
+        stream = self.stream
+        if stream is None or getattr(stream, "closed", False):
+            return
+        try:
+            super().emit(record)
+        except ValueError:
+            return
+
+
+def _configure_default_logging() -> None:
+    logger = logging.getLogger("shared_tensor")
+    if logger.handlers:
+        return
+    level_name = os.getenv("SHARED_TENSOR_LOG_LEVEL", "INFO").strip().upper() or "INFO"
+    level = getattr(logging, level_name, logging.INFO)
+    handler = _SafeStreamHandler()
+    handler.setFormatter(logging.Formatter("[shared_tensor] %(levelname)s %(name)s: %(message)s"))
+    logger.addHandler(handler)
+    logger.setLevel(level)
+    logger.propagate = False
+
+_configure_default_logging()
+
+from shared_tensor.async_client import AsyncSharedTensorClient
+from shared_tensor.async_task import TaskInfo, TaskStatus
+from shared_tensor.client import SharedTensorClient
+from shared_tensor.errors import SharedTensorStaleHandleError
+from shared_tensor.managed_object import SharedObjectHandle
+from shared_tensor.provider import SharedTensorProvider
+from shared_tensor.server import SharedTensorServer
+
+__all__ = [
+    "AsyncSharedTensorClient",
+    "SharedTensorClient",
+    "SharedObjectHandle",
+    "SharedTensorStaleHandleError",
+    "SharedTensorProvider",
+    "SharedTensorServer",
+    "TaskInfo",
+    "TaskStatus",
+]
+
+__version__ = "0.2.16"

{shared_tensor-0.2.15 → shared_tensor-0.2.16}/shared_tensor/async_client.py

@@ -20,7 +20,7 @@ class AsyncSharedTensorClient:
     def __init__(
         self,
         base_path: str = "/tmp/shared-tensor",
-        verbose_debug: bool = False,
+        verbose_debug: bool = True,
         poll_interval: float = 1.0,
         *,
         device_index: int | None = None,

{shared_tensor-0.2.15 → shared_tensor-0.2.16}/shared_tensor/client.py

@@ -59,7 +59,7 @@ class SharedTensorClient:
         *,
         device_index: int | None = None,
         timeout: float = 600.0,
-        verbose_debug: bool = False,
+        verbose_debug: bool = True,
     ) -> None:
         self.base_path = base_path
         self.device_index = device_index

{shared_tensor-0.2.15 → shared_tensor-0.2.16}/shared_tensor/provider.py

@@ -94,7 +94,7 @@ class SharedTensorProvider:
         timeout: float = 600.0,
         execution_mode: str = "auto",
         server_startup_timeout: float = 30.0,
-        verbose_debug: bool = False,
+        verbose_debug: bool = True,
     ) -> None:
         resolved_mode, auto_mode = _resolve_execution_mode(
             execution_mode,

{shared_tensor-0.2.15 → shared_tensor-0.2.16}/shared_tensor/server.py

@@ -97,7 +97,7 @@ class SharedTensorServer:
         result_ttl: float = 3600.0,
         process_start_method: str | None = None,
         startup_timeout: float = 30.0,
-        verbose_debug: bool = False,
+        verbose_debug: bool = True,
     ) -> None:
         self.provider = provider or SharedTensorProvider(execution_mode="server")
         self.socket_path = socket_path or resolve_runtime_socket_path(

shared_tensor-0.2.15/shared_tensor/__init__.py

@@ -1,22 +0,0 @@
-"""shared_tensor: same-host same-GPU PyTorch CUDA IPC over local UDS RPC."""
-
-from shared_tensor.async_client import AsyncSharedTensorClient
-from shared_tensor.async_task import TaskInfo, TaskStatus
-from shared_tensor.client import SharedTensorClient
-from shared_tensor.errors import SharedTensorStaleHandleError
-from shared_tensor.managed_object import SharedObjectHandle
-from shared_tensor.provider import SharedTensorProvider
-from shared_tensor.server import SharedTensorServer
-
-__all__ = [
-    "AsyncSharedTensorClient",
-    "SharedTensorClient",
-    "SharedObjectHandle",
-    "SharedTensorStaleHandleError",
-    "SharedTensorProvider",
-    "SharedTensorServer",
-    "TaskInfo",
-    "TaskStatus",
-]
-
-__version__ = "0.2.15"