Skip to content

feat(experimentalist): add Tau3 experimentalist workflow to docs - #999

Merged
nicot merged 1 commit into
mainfrom
tau3-experimentalist-cli-example/ntonozzi
Jul 30, 2026
Merged

feat(experimentalist): add Tau3 experimentalist workflow to docs#999
nicot merged 1 commit into
mainfrom
tau3-experimentalist-cli-example/ntonozzi

Conversation

@nicot

@nicot nicot commented Jul 30, 2026

Copy link
Copy Markdown
Contributor

Summary

  • add a small official Tau3 Airline train/validation dataset preparation script
  • Add docs for running the experimentalist smoke test
    This is not a great UX yet -- my next step will be to go back and clean up the docs and remove tau2, but this is working for now and I want to let others test it out.

Summary by CodeRabbit

  • New Features

    • Added a Tau3 Airline smoke-dataset preparation workflow for training and validation.
    • Added a fast smoke-run configuration for Tau3 agent experiments.
    • Added setup instructions for running the optimization workflow end-to-end and reviewing results.
  • Bug Fixes

    • Improved model selection so the configured model is consistently passed into the agent environment.

@nicot
nicot requested review from a team as code owners July 30, 2026 20:06
@github-actions github-actions Bot added the feat label Jul 30, 2026
@github-actions

Copy link
Copy Markdown
Contributor

@coderabbitai

coderabbitai Bot commented Jul 30, 2026

Copy link
Copy Markdown
Contributor

Review Change Stack

📝 Walkthrough

Walkthrough

The PR documents an end-to-end τ-Bench optimization example, adds Tau3 Airline smoke dataset preparation and bounded Experimentalist settings, and propagates the configured or environment-provided model into the agent container.

Changes

Tau3 Airline smoke workflow

Layer / File(s) Summary
Smoke dataset preparation
plugins/nemo-experimentalist/examples/tau3-nooa-agent/prepare-airline-smoke.sh
Downloads Tau3 data when needed, validates existing splits, copies the expected train and validation tasks, and reports their paths.
Bounded Experimentalist run
docs/get-started/example-agent.mdx, plugins/nemo-experimentalist/examples/tau3-nooa-agent/experimentalist-smoke.yaml
Documents environment setup, workspace creation, smoke dataset preparation, bounded optimization execution, and result inspection using the new smoke configuration.
Model environment fallback
plugins/nemo-experimentalist/examples/tau3-nooa-agent/harbor_wrapper.py
Sets NEMO_AGENT_MODEL from self.model_name or the AUT_MODEL_NAME environment variable.

Sequence Diagram(s)

sequenceDiagram
  participant User
  participant prepare-airline-smoke.sh
  participant Experimentalist
  participant Tau3 NOOA agent
  User->>prepare-airline-smoke.sh: Prepare train and validation smoke datasets
  prepare-airline-smoke.sh-->>User: Return dataset paths
  User->>Experimentalist: Start bounded optimization run
  Experimentalist->>Tau3 NOOA agent: Launch candidate with model environment
  Tau3 NOOA agent-->>Experimentalist: Return evaluation results
  Experimentalist-->>User: Write selected winner to run.json
Loading

Possibly related PRs

Suggested labels: docs

Suggested reviewers: aleckhoury, callingmedic911, tylersbray

🚥 Pre-merge checks | ✅ 5
✅ Passed checks (5 passed)
Check name Status Explanation
Docstring Coverage ✅ Passed No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Linked Issues check ✅ Passed Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check ✅ Passed Check skipped because no linked issues were found for this pull request.
Description Check ✅ Passed Check skipped - CodeRabbit’s high-level summary is enabled.
Title check ✅ Passed The title matches the main change: adding a Tau3 experimentalist workflow and related documentation/configuration.
✨ Finishing Touches
📝 Generate docstrings
  • Create stacked PR
  • Commit on current branch
🧪 Generate unit tests (beta)
  • Create PR with unit tests
  • Commit unit tests in branch tau3-experimentalist-cli-example/ntonozzi

Comment @coderabbitai help to get the list of available commands.

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 1

🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In
`@plugins/nemo-experimentalist/examples/tau3-nooa-agent/prepare-airline-smoke.sh`:
- Around line 27-36: Update the dataset cache guard in prepare-airline-smoke.sh
to verify that every selected task in TRAIN_TASKS has its task.toml under
DATASET_ROOT before skipping the download. Redownload via the existing uv run
harbor download block whenever any of the five tasks is missing, while
preserving reuse only when all task files exist.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: 6ed37ac1-9909-4a5c-9f06-347b3fe6b82d

📥 Commits

Reviewing files that changed from the base of the PR and between 9b0f9ad and 0045319.

📒 Files selected for processing (6)
  • docs/get-started/example-agent.mdx
  • plugins/nemo-experimentalist/examples/tau3-nooa-agent/README.md
  • plugins/nemo-experimentalist/examples/tau3-nooa-agent/experimentalist-smoke.yaml
  • plugins/nemo-experimentalist/examples/tau3-nooa-agent/harbor_wrapper.py
  • plugins/nemo-experimentalist/examples/tau3-nooa-agent/prepare-airline-smoke.sh
  • plugins/nemo-experimentalist/tests/experimentalist/test_tau3_nooa_agent.py

@github-actions

github-actions Bot commented Jul 30, 2026

Copy link
Copy Markdown
Contributor
Suite Lines Covered Line Rate Branch Rate
Unit Tests 28406/36247 78.4% 62.8%
Integration Tests 16988/34965 48.6% 21.2%

Provide a bounded official-dataset workflow that customers can run through the NeMo CLI while recording Platform traces.

Signed-off-by: Nico Tonozzi <ntonozzi@nvidia.com>
@nicot
nicot force-pushed the tau3-experimentalist-cli-example/ntonozzi branch from 0045319 to 50dd92c Compare July 30, 2026 20:33
@nicot nicot changed the title feat(experimentalist): add Tau3 Airline CLI smoke workflow feat(experimentalist): add Tau3 experimentalist workflow to docs Jul 30, 2026

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 1

🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@docs/get-started/example-agent.mdx`:
- Line 88: Correct the spelling in the agent performance sentence by changing
“evalaution” to “evaluation,” without altering the surrounding text.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: 7c0cfd0c-e4c0-4629-b9f7-23447c0600dc

📥 Commits

Reviewing files that changed from the base of the PR and between 0045319 and 50dd92c.

📒 Files selected for processing (4)
  • docs/get-started/example-agent.mdx
  • plugins/nemo-experimentalist/examples/tau3-nooa-agent/experimentalist-smoke.yaml
  • plugins/nemo-experimentalist/examples/tau3-nooa-agent/harbor_wrapper.py
  • plugins/nemo-experimentalist/examples/tau3-nooa-agent/prepare-airline-smoke.sh

Comment thread docs/get-started/example-agent.mdx
@nicot
nicot added this pull request to the merge queue Jul 30, 2026
Merged via the queue into main with commit fef4a5d Jul 30, 2026
66 checks passed
@nicot
nicot deleted the tau3-experimentalist-cli-example/ntonozzi branch July 30, 2026 22:07
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants