tetra-rp 0.20.0__tar.gz → 0.20.1__tar.gz

{tetra_rp-0.20.0 → tetra_rp-0.20.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tetra_rp
-Version: 0.20.0
+Version: 0.20.1
 Summary: A Python library for distributed inference and serving of machine learning models
 Author-email: Marut Pandya <pandyamarut@gmail.com>, Patrick Rachford <prachford@icloud.com>, Dean Quinanola <dean.quinanola@runpod.io>
 License: MIT
@@ -8,7 +8,7 @@ Classifier: Development Status :: 3 - Alpha
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
-Requires-Python: <3.14,>=3.9
+Requires-Python: <3.15,>=3.10
 Description-Content-Type: text/markdown
 Requires-Dist: cloudpickle>=3.1.1
 Requires-Dist: runpod

{tetra_rp-0.20.0 → tetra_rp-0.20.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tetra_rp"
-version = "0.20.0"
+version = "0.20.1"
 description = "A Python library for distributed inference and serving of machine learning models"
 authors = [
     { name = "Marut Pandya", email = "pandyamarut@gmail.com" },
@@ -15,7 +15,7 @@ classifiers = [
     "License :: OSI Approved :: MIT License",
     "Operating System :: OS Independent",
 ]
-requires-python = ">=3.9,<3.14"
+requires-python = ">=3.10,<3.15"
 
 dependencies = [
     "cloudpickle>=3.1.1",
@@ -39,6 +39,7 @@ test = [
     "pytest-mock>=3.14.0",
     "pytest-asyncio>=1.0.0",
     "pytest-cov>=6.2.1",
+    "pytest-xdist>=3.6.1",
     "twine>=6.1.0",
 ]
 
@@ -68,14 +69,15 @@ addopts = [
     "--tb=short",
     "--cov=tetra_rp",
     "--cov-report=term-missing",
-    "--cov-fail-under=35"
+    "--cov-fail-under=65"
 ]
 asyncio_mode = "auto"
 asyncio_default_fixture_loop_scope = "function"
 markers = [
     "unit: Unit tests",
     "integration: Integration tests",
-    "slow: Slow tests"
+    "slow: Slow tests",
+    "serial: Tests that must run serially (not parallelized)"
 ]
 filterwarnings = [
     "ignore::DeprecationWarning",
@@ -92,7 +94,7 @@ exclude = [
 
 [tool.mypy]
 # Basic configuration
-python_version = "3.9"
+python_version = "3.10"
 warn_return_any = true
 warn_unused_configs = true
 disallow_untyped_defs = false  # Start lenient, can be stricter later
@@ -122,5 +124,36 @@ module = [
 ]
 ignore_missing_imports = true
 
+[tool.coverage.run]
+parallel = true
+branch = false
+source = ["tetra_rp"]
+omit = [
+    "*/tests/*",
+    "*/test_*.py",
+    "*/__pycache__/*",
+    "*/site-packages/*",
+]
+
+[tool.coverage.report]
+precision = 2
+show_missing = true
+skip_covered = false
+exclude_lines = [
+    "pragma: no cover",
+    "def __repr__",
+    "raise AssertionError",
+    "raise NotImplementedError",
+    "if __name__ == .__main__.:",
+    "if TYPE_CHECKING:",
+    "@abstractmethod",
+]
+
+[tool.coverage.paths]
+source = [
+    "src/tetra_rp",
+    "*/site-packages/tetra_rp",
+]
+
 [tool.uv.sources]
 runpod = { git = "https://github.com/runpod/runpod-python", rev = "main" }

tetra_rp-0.20.1/src/tetra_rp/core/resources/template.py

@@ -0,0 +1,36 @@
+from typing import Dict, List, Optional, Any
+from pydantic import BaseModel, model_validator
+from .base import BaseResource
+
+
+class KeyValuePair(BaseModel):
+    key: str
+    value: str
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, str]) -> "List[KeyValuePair]":
+        """
+        Create a list of KeyValuePair instances from a dictionary.
+        """
+        if not isinstance(data, dict):
+            raise ValueError("Input must be a dictionary.")
+
+        return [cls(key=key, value=value) for key, value in data.items()]
+
+
+class PodTemplate(BaseResource):
+    advancedStart: Optional[bool] = False
+    config: Optional[Dict[str, Any]] = {}
+    containerDiskInGb: Optional[int] = 64
+    containerRegistryAuthId: Optional[str] = ""
+    dockerArgs: Optional[str] = ""
+    env: Optional[List[KeyValuePair]] = []
+    imageName: Optional[str] = ""
+    name: Optional[str] = ""
+    ports: Optional[str] = ""
+    startScript: Optional[str] = ""
+
+    @model_validator(mode="after")
+    def sync_input_fields(self):
+        self.name = f"{self.name}__{self.resource_id}"
+        return self

tetra_rp-0.20.1/src/tetra_rp/core/utils/json.py

@@ -0,0 +1,46 @@
+"""Utilities for normalizing data structures for JSON serialization."""
+
+from enum import Enum
+from typing import Any
+
+from pydantic import BaseModel
+
+
+def normalize_for_json(obj: Any) -> Any:
+    """Normalize an object for JSON serialization.
+
+    Converts Pydantic models to dicts and Enum values to their values,
+    while recursively processing collections.
+
+    Args:
+        obj: The object to normalize.
+
+    Returns:
+        A JSON-serializable version of the object.
+    """
+    # Handle primitives
+    if obj is None or isinstance(obj, (bool, int, float, str)):
+        return obj
+
+    # Handle Enum
+    if isinstance(obj, Enum):
+        return obj.value
+
+    # Handle Pydantic BaseModel
+    if isinstance(obj, BaseModel):
+        return normalize_for_json(obj.model_dump())
+
+    # Handle dict
+    if isinstance(obj, dict):
+        return {key: normalize_for_json(value) for key, value in obj.items()}
+
+    # Handle tuple
+    if isinstance(obj, tuple):
+        return tuple(normalize_for_json(item) for item in obj)
+
+    # Handle list
+    if isinstance(obj, list):
+        return [normalize_for_json(item) for item in obj]
+
+    # For any other type, return as-is
+    return obj

{tetra_rp-0.20.0 → tetra_rp-0.20.1}/src/tetra_rp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tetra_rp
-Version: 0.20.0
+Version: 0.20.1
 Summary: A Python library for distributed inference and serving of machine learning models
 Author-email: Marut Pandya <pandyamarut@gmail.com>, Patrick Rachford <prachford@icloud.com>, Dean Quinanola <dean.quinanola@runpod.io>
 License: MIT
@@ -8,7 +8,7 @@ Classifier: Development Status :: 3 - Alpha
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
-Requires-Python: <3.14,>=3.9
+Requires-Python: <3.15,>=3.10
 Description-Content-Type: text/markdown
 Requires-Dist: cloudpickle>=3.1.1
 Requires-Dist: runpod

{tetra_rp-0.20.0 → tetra_rp-0.20.1}/src/tetra_rp.egg-info/SOURCES.txt

@@ -66,7 +66,6 @@ src/tetra_rp/core/resources/resource_manager.py
 src/tetra_rp/core/resources/serverless.py
 src/tetra_rp/core/resources/serverless_cpu.py
 src/tetra_rp/core/resources/template.py
-src/tetra_rp/core/resources/utils.py
 src/tetra_rp/core/utils/__init__.py
 src/tetra_rp/core/utils/backoff.py
 src/tetra_rp/core/utils/constants.py

tetra_rp-0.20.0/src/tetra_rp/core/resources/template.py

@@ -1,107 +0,0 @@
-import warnings
-from typing import Dict, List, Optional, Any
-from pydantic import BaseModel, model_validator
-from tetra_rp.core.utils.http import get_authenticated_requests_session
-from .base import BaseResource
-
-
-class KeyValuePair(BaseModel):
-    key: str
-    value: str
-
-    @classmethod
-    def from_dict(cls, data: Dict[str, str]) -> "List[KeyValuePair]":
-        """
-        Create a list of KeyValuePair instances from a dictionary.
-        """
-        if not isinstance(data, dict):
-            raise ValueError("Input must be a dictionary.")
-
-        return [cls(key=key, value=value) for key, value in data.items()]
-
-
-class PodTemplate(BaseResource):
-    advancedStart: Optional[bool] = False
-    config: Optional[Dict[str, Any]] = {}
-    containerDiskInGb: Optional[int] = 64
-    containerRegistryAuthId: Optional[str] = ""
-    dockerArgs: Optional[str] = ""
-    env: Optional[List[KeyValuePair]] = []
-    imageName: Optional[str] = ""
-    name: Optional[str] = ""
-    ports: Optional[str] = ""
-    startScript: Optional[str] = ""
-
-    @model_validator(mode="after")
-    def sync_input_fields(self):
-        self.name = f"{self.name}__{self.resource_id}"
-        return self
-
-
-def update_system_dependencies(
-    template_id, token=None, system_dependencies=None, base_entry_cmd=None
-):
-    """
-    Updates Runpod template with system dependencies installed via apt-get,
-    and appends the app start command.
-
-    Args:
-        template_id (str): Runpod template ID.
-        token (str): [DEPRECATED] Runpod API token. Ignored; uses RUNPOD_API_KEY env var instead.
-        system_dependencies (List[str]): List of apt packages to install.
-        base_entry_cmd (List[str]): The default command to run the app, e.g. ["uv", "run", "handler.py"]
-    Returns:
-        dict: API response JSON or error info.
-    """
-    # Warn if deprecated token parameter is used
-    if token is not None:
-        warnings.warn(
-            "The 'token' parameter is deprecated and ignored. "
-            "Authentication now uses RUNPOD_API_KEY environment variable.",
-            DeprecationWarning,
-            stacklevel=2,
-        )
-
-    # Compose apt-get install command if any packages specified
-    apt_cmd = ""
-    if system_dependencies:
-        joined_pkgs = " ".join(system_dependencies)
-        apt_cmd = f"apt-get update && apt-get install -y {joined_pkgs} && "
-
-    # Default start command if not provided
-    app_cmd = base_entry_cmd or ["uv", "run", "handler.py"]
-    app_cmd_str = " ".join(app_cmd)
-
-    # Full command to run in entrypoint shell
-    full_cmd = f"{apt_cmd}exec {app_cmd_str}"
-
-    payload = {
-        # other required fields like disk, env, image, etc, should be fetched or passed in real usage
-        "dockerEntrypoint": ["/bin/bash", "-c", full_cmd],
-        "dockerStartCmd": [],
-        # placeholder values, replace as needed or fetch from current template state
-        "containerDiskInGb": 50,
-        "containerRegistryAuthId": "",
-        "env": {},
-        "imageName": "your-image-name",
-        "isPublic": False,
-        "name": "your-template-name",
-        "ports": ["8888/http", "22/tcp"],
-        "readme": "",
-        "volumeInGb": 20,
-        "volumeMountPath": "/workspace",
-    }
-
-    url = f"https://rest.runpod.io/v1/templates/{template_id}/update"
-
-    # Use centralized auth utility instead of manual header setup
-    # Note: token parameter is deprecated; uses RUNPOD_API_KEY environment variable
-    session = get_authenticated_requests_session()
-    try:
-        response = session.post(url, json=payload)
-        response.raise_for_status()
-        return response.json()
-    except Exception as e:
-        return {"error": "Failed to update template", "details": str(e)}
-    finally:
-        session.close()

tetra_rp-0.20.0/src/tetra_rp/core/resources/utils.py

@@ -1,50 +0,0 @@
-from typing import Callable, Any, List, Union
-from pydantic import BaseModel
-from .gpu import GpuType, GpuTypeDetail
-from .serverless import ServerlessEndpoint
-
-
-"""
-Define the mapping for the methods and their return types
-Only include methods from runpod.*
-"""
-RUNPOD_TYPED_OPERATIONS = {
-    "get_gpus": List[GpuType],
-    "get_gpu": GpuTypeDetail,
-    "get_endpoints": List[ServerlessEndpoint],
-}
-
-
-def inquire(method: Callable, *args, **kwargs) -> Union[List[Any], Any]:
-    """
-    This function dynamically determines the return type of the provided method
-    based on a predefined mapping (`definitions`) and validates the result using
-    Pydantic models if applicable.
-
-    Refer to `RUNPOD_TYPED_OPERATIONS` for the mapping.
-
-    Example:
-    ----------
-    >>> import runpod
-    >>> inquire(runpod.get_gpus)
-    [
-        GpuType(id='NVIDIA A100 80GB', displayName='A100 80GB', memoryInGb=80),
-        GpuType(id='NVIDIA A100 40GB', displayName='A100 40GB', memoryInGb=40),
-        GpuType(id='NVIDIA A10', displayName='A10', memoryInGb=24)
-    ]
-    """
-    method_name = method.__name__
-    return_type = RUNPOD_TYPED_OPERATIONS.get(method_name)
-
-    raw_result = method(*args, **kwargs)
-
-    if hasattr(return_type, "__origin__") and return_type.__origin__ is list:
-        # List case
-        model_type = return_type.__args__[0]
-        if issubclass(model_type, BaseModel):
-            return [model_type.model_validate(item) for item in raw_result]
-    elif isinstance(return_type, type) and issubclass(return_type, BaseModel):
-        # Single object case
-        return return_type.model_validate(raw_result)
-    else:
-        raise ValueError(f"Unsupported return type for method '{method_name}'")

tetra_rp-0.20.0/src/tetra_rp/core/utils/json.py

@@ -1,33 +0,0 @@
-from enum import Enum
-from typing import Any
-from pydantic import BaseModel
-
-
-def normalize_for_json(obj: Any) -> Any:
-    """
-    Recursively normalizes an object for JSON serialization.
-
-    This function handles various data types and ensures that objects
-    are converted into JSON-serializable formats. It supports the following:
-    - `BaseModel` instances: Converts them to dictionaries using `model_dump()`.
-    - Dictionaries: Recursively normalizes their values.
-    - Lists: Recursively normalizes their elements.
-    - Tuples: Recursively normalizes their elements and returns a tuple.
-    - Other types: Returns the object as is.
-
-    Args:
-        obj (Any): The object to normalize.
-
-    Returns:
-        Any: A JSON-serializable representation of the input object.
-    """
-    if isinstance(obj, BaseModel):
-        return normalize_for_json(obj.model_dump())
-    elif isinstance(obj, Enum):
-        return obj.value
-    elif isinstance(obj, dict):
-        return {k: normalize_for_json(v) for k, v in obj.items()}
-    elif isinstance(obj, (list, tuple)):
-        return type(obj)(normalize_for_json(i) for i in obj)
-    else:
-        return obj