hud-python 0.6.1__tar.gz → 0.6.2__tar.gz

{hud_python-0.6.1 → hud_python-0.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hud-python
-Version: 0.6.1
+Version: 0.6.2
 Summary: SDK for the HUD platform.
 Project-URL: Homepage, https://github.com/hud-evals/hud-python
 Project-URL: Bug Tracker, https://github.com/hud-evals/hud-python/issues

{hud_python-0.6.1 → hud_python-0.6.2}/cookbooks/rl-training/README.md

@@ -18,12 +18,13 @@ each `optim_step` closes the on-policy loop.
 
 ## Run
 
-Needs `HUD_API_KEY` (from your environment or `.env`). List the trainable
-gateway models on your account, pick one, and set it as the `MODEL` constant at
-the top of `simple_train.py` / `ppo_custom_loss.py`:
+Needs `HUD_API_KEY` (from your environment or `.env`). List the gateway models
+on your account, pick a trainable one (the **Trainable** column marks them), and
+set it as the `MODEL` constant at the top of `simple_train.py` /
+`ppo_custom_loss.py`:
 
 ```bash
-hud models
+hud models list          # Name | Model (API) | ID | Provider | Agent | Trainable
 ```
 
 **Train on a deployed taskset (the real flow).** You've built a taskset and

{hud_python-0.6.1 → hud_python-0.6.2}/hud/cli/models.py

@@ -56,14 +56,18 @@ def list_models(
     table = Table()
     table.add_column("Name", style="cyan")
     table.add_column("Model (API)", style="green")
+    table.add_column("ID", style="blue", no_wrap=True)
     table.add_column("Provider", style="yellow")
     table.add_column("Agent", style="magenta")
+    table.add_column("Trainable", style="green", justify="center")
     for model in models_list:
         table.add_row(
             model.name or model.id or "-",
             model.model_name or model.id or "-",
+            model.id or "-",
             model.provider.name or "-",
             model.sdk_agent_type or "-",
+            "✓" if model.is_trainable else "",
         )
     console.print(table)
     console.print(f"\n[dim]Gateway: {settings.hud_gateway_url}[/dim]")

{hud_python-0.6.1 → hud_python-0.6.2}/hud/utils/gateway.py

@@ -35,6 +35,7 @@ class GatewayModelInfo(BaseModel):
     name: str | None = None
     model_name: str | None = None
     sdk_agent_type: str | None = None
+    is_trainable: bool = False
     provider: GatewayProviderInfo = Field(default_factory=GatewayProviderInfo)
 
 

{hud_python-0.6.1 → hud_python-0.6.2}/hud/version.py

@@ -4,4 +4,4 @@ Version information for the HUD SDK.
 
 from __future__ import annotations
 
-__version__ = "0.6.1"
+__version__ = "0.6.2"

{hud_python-0.6.1 → hud_python-0.6.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hud-python"
-version = "0.6.1"
+version = "0.6.2"
 description = "SDK for the HUD platform."
 readme = "README.md"
 requires-python = ">=3.11, <3.13"