hud-python 0.4.41__tar.gz → 0.4.42__tar.gz

{hud_python-0.4.41 → hud_python-0.4.42}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hud-python
-Version: 0.4.41
+Version: 0.4.42
 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.4.41 → hud_python-0.4.42}/hud/cli/rl/gpu_utils.py

@@ -253,6 +253,8 @@ def adjust_config_for_ddp(config: Config, num_gpus: int) -> Config:
     # Update max_parallel_episodes to match
     config.actor.max_parallel_episodes = config.training.batch_size
 
+    config.training.num_gpus = num_gpus
+
     # Log the adjustment
     from rich.console import Console
 

{hud_python-0.4.41 → hud_python-0.4.42}/hud/cli/rl/remote_runner.py

@@ -310,7 +310,7 @@ def run_remote_training(
         # console.print(gpu_table)
 
         if yes:
-            gpu_choice = "A100"  # Default GPU in yes mode
+            gpu_choice = "A100"
             hud_console.info(f"Auto-selecting GPU: {gpu_choice} 80GB (--yes mode)")
         else:
             gpu_choice = hud_console.select(
@@ -323,7 +323,7 @@ def run_remote_training(
             )
 
         if yes:
-            num_gpus = 1  # Default to 1 GPU in yes mode
+            num_gpus = 2 # Default to 2 GPUs in yes mode
             hud_console.info(f"Auto-selecting {num_gpus} GPU(s) (--yes mode)")
         else:
             num_gpus = hud_console.select(
@@ -350,6 +350,8 @@ def run_remote_training(
 
         config = adjust_config_for_ddp(config, int(num_gpus))
 
+        config.training.gpu_type = gpu_choice
+
         # Use a short label for tasks (avoid full absolute paths)
         try:
             if tasks_file and Path(tasks_file).exists():
@@ -424,8 +426,8 @@ def run_remote_training(
         hud_console.info(f"Loading configuration from: {config_file}")
         config = load_config(config_file)
         config_dict = config.to_dict()
-        gpu_choice = "A100"  # Default
-        num_gpus = 1  # Default for non-interactive mode
+        gpu_choice = config.training.gpu_type
+        num_gpus = config.training.num_gpus
 
     # Launch training
     try:

{hud_python-0.4.41 → hud_python-0.4.42}/hud/rl/actor.py

@@ -109,7 +109,7 @@ class Actor:
 
         # Run the task
         try:
-            with hud.trace(f"Training | {task.id}", job_id=job_id):
+            with hud.trace(f"Training | {task.prompt}", job_id=job_id):
                 result = await agent.run(task, max_steps=self.actor_config.max_steps_per_episode)
 
         except Exception:

{hud_python-0.4.41 → hud_python-0.4.42}/hud/rl/config.py

@@ -61,6 +61,9 @@ class ModelConfig:
 @dataclass
 class TrainingConfig:
     """Training hyperparameters."""
+    # GPU parameters
+    gpu_type: str = "A100"
+    num_gpus: int = 2
 
     # Training parameters
     training_steps: int = 100

{hud_python-0.4.41 → hud_python-0.4.42}/hud/rl/train.py

@@ -103,7 +103,10 @@ async def train(config: Config, tasks: list[Task]) -> None:
     if is_main_process():
         hud_console.info(f"Creating job with config.job_id: {config.job_id}")
         job_obj = hud.create_job(
-            job_id=config.job_id, name=config.job_name, metadata={"config": config.to_dict()}
+            job_id=config.job_id, name=config.job_name, metadata={
+                "config": config.to_dict(),
+                "agent_class": config.model.base_model
+            }
         )
         hud_console.info(f"Created job with job_obj.id: {job_obj.id}")
         job_obj.update_status_sync("running")

{hud_python-0.4.41 → hud_python-0.4.42}/hud/utils/tests/test_version.py

@@ -5,4 +5,4 @@ def test_import():
     """Test that the package can be imported."""
     import hud
 
-    assert hud.__version__ == "0.4.41"
+    assert hud.__version__ == "0.4.42"

{hud_python-0.4.41 → hud_python-0.4.42}/hud/version.py

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

{hud_python-0.4.41 → hud_python-0.4.42}/pyproject.toml

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