oddish run shows only new trials

Author: RishiDesai

backend/api/routers/tasks.py

@@ -261,26 +261,26 @@ async def create_task_sweep(
 
         await session.commit()
 
-        provider_counts: Counter[str] = Counter(
-            t.provider for t in (new_trials if is_append else task.trials)
-        )
+        response_trials = new_trials if is_append else list(task.trials)
+        provider_counts: Counter[str] = Counter(t.provider for t in response_trials)
         primary = (
             experiment
             or (task.experiments[0] if task.experiments else None)
         )
         resp_experiment_id = primary.id if primary else None
         resp_experiment_name = primary.name if primary else None
-        
+
         return TaskResponse(
             id=task.id,
             name=task.name,
             status=task.status,
             priority=task.priority,
-            trials_count=len(new_trials) if is_append else len(task.trials),
+            trials_count=len(response_trials),
             providers=dict(provider_counts),
             experiment_id=resp_experiment_id,
             experiment_name=resp_experiment_name,
             created_at=task.created_at,
+            new_trial_ids=[t.id for t in response_trials],
         )
 
 

oddish/src/oddish/cli/api.py

@@ -9,6 +9,7 @@
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from datetime import datetime
 from pathlib import Path
+from collections.abc import Iterable
 from typing import Any, cast
 
 import httpx
@@ -1346,14 +1347,20 @@ def watch_task(
     api_url: str,
     task_id: str,
     experiment_id: str | None = None,
+    trial_ids: Iterable[str] | None = None,
 ) -> dict | None:
     """Watch a task until completion. Returns the final result.
 
     When *experiment_id* is given, only trials belonging to that experiment
     are displayed (others are hidden from the table and summary counts).
+
+    When *trial_ids* is given, only trials whose ``id`` is in that set are
+    shown. This is useful when appending trials to an existing task and the
+    caller only wants to monitor the freshly-submitted trials.
     """
     final_result = None
     headers = get_auth_headers()
+    trial_id_filter = set(trial_ids) if trial_ids is not None else None
     with Live(console=console, refresh_per_second=2) as live:
         while True:
             try:
@@ -1368,7 +1375,11 @@ def watch_task(
                 final_result = result
 
                 all_trials = result.get("trials", [])
-                if experiment_id:
+                if trial_id_filter is not None:
+                    all_trials = [
+                        t for t in all_trials if t.get("id") in trial_id_filter
+                    ]
+                elif experiment_id:
                     all_trials = [
                         t for t in all_trials if t.get("experiment_id") == experiment_id
                     ]
@@ -1450,7 +1461,7 @@ def watch_task(
                 live.update(table)
 
                 # Check if done
-                if experiment_id:
+                if trial_id_filter is not None or experiment_id:
                     terminal = {"success", "failed", "cancelled"}
                     if all_trials and all(
                         t.get("status") in terminal for t in all_trials

oddish/src/oddish/cli/run.py

@@ -715,11 +715,17 @@ def submit_task(
         if not quiet:
             console.print("[dim]Watching task progress (Ctrl+C to stop)...[/dim]")
             console.print()
+        # When appending to an existing task, restrict the live view to the
+        # trials we just submitted so prior trials on the same experiment
+        # don't clutter the table. For fresh tasks the list is equivalent
+        # to the full trial set anyway, so passing it is harmless.
+        new_trial_ids = all_results[0].get("new_trial_ids") or None
         try:
             final_result = watch_task(
                 api_url,
                 all_results[0]["id"],
                 experiment_id=experiment_id_resolved,
+                trial_ids=new_trial_ids,
             )
             # Print final results table
             if final_result:

oddish/src/oddish/schemas.py

@@ -463,6 +463,16 @@ class TaskResponse(BaseModel):
     experiment_id: str | None = None
     experiment_name: str | None = None
     created_at: datetime
+    new_trial_ids: list[str] = Field(
+        default_factory=list,
+        description=(
+            "IDs of the trials created by this sweep submission. "
+            "For append-mode submissions, this contains only the newly appended "
+            "trials (not any pre-existing trials on the task). Clients can use "
+            "this to filter status/watch views to only the trials they just "
+            "submitted."
+        ),
+    )
 
 
 class TaskBatchCancelRequest(BaseModel):

oddish/src/oddish/server/__init__.py

@@ -361,26 +361,26 @@ async def create_task_sweep(submission: TaskSweepSubmission):
         if not is_append and hasattr(task, "task_s3_key") and task.task_s3_key:
             await session.commit()
 
-        provider_counts: Counter[str] = Counter(
-            t.provider for t in (new_trials if is_append else task.trials)
-        )
+        response_trials = new_trials if is_append else list(task.trials)
+        provider_counts: Counter[str] = Counter(t.provider for t in response_trials)
         primary = (
             experiment
             or (task.experiments[0] if task.experiments else None)
         )
         resp_experiment_id = primary.id if primary else None
         resp_experiment_name = primary.name if primary else None
-        
+
         return TaskResponse(
             id=task.id,
             name=task.name,
             status=task.status,
             priority=task.priority,
-            trials_count=len(new_trials) if is_append else len(task.trials),
+            trials_count=len(response_trials),
             providers=dict(provider_counts),
             experiment_id=resp_experiment_id,
             experiment_name=resp_experiment_name,
             created_at=task.created_at,
+            new_trial_ids=[t.id for t in response_trials],
         )