baseten-loops-tinker 0.1.0__tar.gz

baseten_loops_tinker-0.1.0/.gitignore

@@ -0,0 +1,10 @@
+.env
+__pycache__/
+*.pyc
+*.pyo
+.venv/
+dist/
+*.egg-info/
+.pytest_cache/
+.mypy_cache/
+.ruff_cache/

baseten_loops_tinker-0.1.0/PKG-INFO

@@ -0,0 +1,71 @@
+Metadata-Version: 2.4
+Name: baseten-loops-tinker
+Version: 0.1.0
+Summary: Drop-in tinker compatibility shim backed by the loops SDK
+Requires-Python: >=3.13
+Requires-Dist: baseten-loops>=0.1.0
+Description-Content-Type: text/markdown
+
+# loops-tinker
+
+Drop-in compatibility shim that provides the `tinker` package namespace backed
+by the `loops` SDK.
+
+## Installation
+
+```bash
+pip install loops-tinker
+```
+
+## Usage
+
+Existing code that uses `import tinker` works without modification:
+
+```python
+import tinker
+
+service_client = tinker.ServiceClient(training_project_id="proj_abc123")
+training_client = service_client.create_lora_training_client(
+    base_model="Qwen/Qwen3-8B", rank=16,
+)
+sampling_client = service_client.create_sampling_client(model="Qwen/Qwen3-8B")
+```
+
+All types are also available under `tinker.types`:
+
+```python
+from tinker.types import SamplingParams, Datum, ModelInput
+from tinker.types.tensor_data import TensorData
+```
+
+## Using with tinker-cookbook
+
+`tinker-cookbook` depends on `tinker`, but you can swap in `loops-tinker` using
+uv's `override-dependencies` to prevent the original `tinker` from being installed.
+
+First add the override to `pyproject.toml` **before** installing any dependencies
+(there is no CLI command for this). If `tinker-cookbook` is added first, the
+original `tinker` gets installed and its files conflict with `loops-tinker`:
+
+```toml
+[tool.uv]
+override-dependencies = [
+    "tinker ; python_version < '0'",
+]
+```
+
+Then add the dependencies:
+
+```bash
+uv init
+uv add tinker-cookbook
+uv add "loops @ git+https://github.com/basetenlabs/loops#subdirectory=loops"
+uv add "loops-tinker @ git+https://github.com/basetenlabs/loops#subdirectory=loops-tinker"
+```
+
+`loops` is a transitive dep of `loops-tinker` but must be listed directly
+because `[tool.uv.sources]` only applies to direct dependencies. **This won't
+be needed once `loops` is published to PyPI.**
+
+The `tinker` namespace will be provided by `loops-tinker` instead of the
+original package.

baseten_loops_tinker-0.1.0/README.md

@@ -0,0 +1,63 @@
+# loops-tinker
+
+Drop-in compatibility shim that provides the `tinker` package namespace backed
+by the `loops` SDK.
+
+## Installation
+
+```bash
+pip install loops-tinker
+```
+
+## Usage
+
+Existing code that uses `import tinker` works without modification:
+
+```python
+import tinker
+
+service_client = tinker.ServiceClient(training_project_id="proj_abc123")
+training_client = service_client.create_lora_training_client(
+    base_model="Qwen/Qwen3-8B", rank=16,
+)
+sampling_client = service_client.create_sampling_client(model="Qwen/Qwen3-8B")
+```
+
+All types are also available under `tinker.types`:
+
+```python
+from tinker.types import SamplingParams, Datum, ModelInput
+from tinker.types.tensor_data import TensorData
+```
+
+## Using with tinker-cookbook
+
+`tinker-cookbook` depends on `tinker`, but you can swap in `loops-tinker` using
+uv's `override-dependencies` to prevent the original `tinker` from being installed.
+
+First add the override to `pyproject.toml` **before** installing any dependencies
+(there is no CLI command for this). If `tinker-cookbook` is added first, the
+original `tinker` gets installed and its files conflict with `loops-tinker`:
+
+```toml
+[tool.uv]
+override-dependencies = [
+    "tinker ; python_version < '0'",
+]
+```
+
+Then add the dependencies:
+
+```bash
+uv init
+uv add tinker-cookbook
+uv add "loops @ git+https://github.com/basetenlabs/loops#subdirectory=loops"
+uv add "loops-tinker @ git+https://github.com/basetenlabs/loops#subdirectory=loops-tinker"
+```
+
+`loops` is a transitive dep of `loops-tinker` but must be listed directly
+because `[tool.uv.sources]` only applies to direct dependencies. **This won't
+be needed once `loops` is published to PyPI.**
+
+The `tinker` namespace will be provided by `loops-tinker` instead of the
+original package.

baseten_loops_tinker-0.1.0/pyproject.toml

@@ -0,0 +1,16 @@
+[project]
+name = "baseten-loops-tinker"
+version = "0.1.0"
+description = "Drop-in tinker compatibility shim backed by the loops SDK"
+readme = "README.md"
+requires-python = ">=3.13"
+dependencies = [
+    "baseten-loops>=0.1.0",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["tinker"]

baseten_loops_tinker-0.1.0/tinker/__init__.py

@@ -0,0 +1,24 @@
+"""Tinker compatibility shim — delegates to the loops SDK.
+
+Install ``loops-tinker`` to use existing ``import tinker`` code with the
+loops backend.  This package re-exports the loops public API under the
+``tinker`` namespace so that tinker-cookbook recipes and other tinker-based
+scripts work without modification.
+"""
+
+from . import types  # noqa: F401 — re-export the types subpackage
+from .types import StopReason  # noqa: F401
+from ._errors import (  # noqa: F401
+    TinkerError,
+    APIError,
+    APIConnectionError,
+    APITimeoutError,
+    AuthenticationError,
+    BadRequestError,
+    RateLimitError,
+)
+from ._service_client import ServiceClient
+from baseten.loops.training_client import TrainingClient  # noqa: F401
+from baseten.loops.training_client import OperationFuture as APIFuture
+from baseten.loops.sampling_client import SamplingClient
+from baseten.loops.models import *  # noqa: F403

baseten_loops_tinker-0.1.0/tinker/_errors.py

@@ -0,0 +1,34 @@
+"""Tinker-compatible exception hierarchy.
+
+These mirror the upstream tinker SDK exception classes so that cookbook
+code like ``except tinker.TinkerError`` or ``isinstance(e, tinker.BadRequestError)``
+works against loops-tinker.
+"""
+
+
+class TinkerError(Exception):
+    """Base class for all tinker/loops errors."""
+
+
+class APIError(TinkerError):
+    """An error returned by the API."""
+
+
+class APIConnectionError(APIError):
+    """Could not connect to the API."""
+
+
+class APITimeoutError(APIConnectionError):
+    """Request timed out."""
+
+
+class AuthenticationError(APIError):
+    """Invalid API key or credentials."""
+
+
+class BadRequestError(APIError):
+    """The request was malformed."""
+
+
+class RateLimitError(APIError):
+    """Rate limit exceeded."""

baseten_loops_tinker-0.1.0/tinker/_service_client.py

@@ -0,0 +1,36 @@
+"""Tinker-compatible ServiceClient adapter.
+
+Adds ``_async`` method variants that the cookbook expects. These currently
+delegate to the synchronous loops implementations.
+"""
+
+from __future__ import annotations
+
+import sys
+from typing import Any
+
+from baseten.loops.service_client import ServiceClient as _LoopsServiceClient
+from baseten.loops.training_client import TrainingClient
+
+
+def _warn_user_metadata(user_metadata: dict[str, str] | None) -> None:
+    if user_metadata:
+        print(
+            "loops-tinker: user_metadata is not yet supported and will be ignored",
+            file=sys.stderr,
+        )
+
+
+class ServiceClient(_LoopsServiceClient):
+    """ServiceClient with async method variants for tinker-cookbook compatibility."""
+
+    async def create_lora_training_client_async(
+        self,
+        base_model: str,
+        rank: int = 16,
+        *,
+        user_metadata: dict[str, str] | None = None,
+        **kwargs: Any,
+    ) -> TrainingClient:
+        _warn_user_metadata(user_metadata)
+        return self.create_lora_training_client(base_model=base_model, rank=rank, **kwargs)

baseten_loops_tinker-0.1.0/tinker/lib/__init__.py

@@ -0,0 +1 @@
+"""Tinker lib compatibility — stubs for tinker.lib subpackages."""

baseten_loops_tinker-0.1.0/tinker/lib/public_interfaces/__init__.py

@@ -0,0 +1,3 @@
+from baseten.loops.training_client import OperationFuture as APIFuture
+
+__all__ = ["APIFuture"]

baseten_loops_tinker-0.1.0/tinker/lib/public_interfaces.py

@@ -0,0 +1,5 @@
+"""Tinker compatibility — re-exports public interfaces."""
+
+from baseten.loops.training_client import OperationFuture as APIFuture
+
+__all__ = ["APIFuture"]

baseten_loops_tinker-0.1.0/tinker/types/__init__.py

@@ -0,0 +1,22 @@
+"""Tinker types compatibility — re-exports loops model types.
+
+Supports ``from tinker import types`` and ``from tinker.types import X``
+patterns used in tinker-cookbook recipes.
+"""
+
+from __future__ import annotations
+
+from typing import Literal, TypeAlias
+from baseten.loops.models import *  # noqa: F403
+from tinker.types._image_asset_pointer_chunk import ImageAssetPointerChunk  # noqa: F401
+
+# Type aliases present in tinker.types but not modelled as classes in loops.
+LossFnType: TypeAlias = Literal[
+    "cross_entropy",
+    "importance_sampling",
+    "ppo",
+    "cispo",
+    "dro",
+]
+
+StopReason: TypeAlias = Literal["stop", "length"]

baseten_loops_tinker-0.1.0/tinker/types/_image_asset_pointer_chunk.py

@@ -0,0 +1,26 @@
+"""Stub for tinker.types.ImageAssetPointerChunk.
+
+Used by tinker-cookbook in isinstance checks for VLM data truncation.
+Trainers does not have an equivalent type yet.
+"""
+
+from __future__ import annotations
+
+from typing import Literal
+
+from pydantic import BaseModel
+
+
+class ImageAssetPointerChunk(BaseModel):
+    format: Literal["png", "jpeg"]
+    location: str
+    expected_tokens: int | None = None
+    type: Literal["image_asset_pointer"] = "image_asset_pointer"
+
+    @property
+    def length(self) -> int:
+        if self.expected_tokens is None:
+            raise ValueError(
+                "ImageAssetPointerChunk expected_tokens needs to be set in order to compute the length"
+            )
+        return self.expected_tokens

baseten_loops_tinker-0.1.0/tinker/types/tensor_data.py

@@ -0,0 +1,5 @@
+"""Compatibility re-export for ``from tinker.types.tensor_data import TensorData``."""
+
+from baseten.loops.models import TensorData as TensorData
+
+__all__ = ["TensorData"]