Skip to main content

trialdesignbench

PyPI version Python versions CI tests Mypy check Ruff check Documentation License

TrialDesignBench is a community-driven benchmark for evaluating AI agents in clinical trial design. The benchmark currently focuses on two core tasks:

  • Design reproduction: Given a Statistical Analysis Plan (SAP) or study protocol, evaluate how accurately AI agents can reproduce the trial design using R.
  • Design generation: Given high-level clinical requirements, evaluate the ability of AI agents to draft new clinical trial designs using R.

How it works

trialdesignbench is a Python package and a thin evaluation framework. It owns the task schema, task materialization, the grader, scoring rules, aggregation, and provenance.

Harbor is the execution backend that runs agent harnesses in Docker. The two interact only through files: Harbor task directories, a generated job.yaml, and the job directory that Harbor writes. The currently supported agents are:

  • Claude Code (claude-code)
  • Codex CLI (codex)
  • Grok Build (grok-build)
  • OpenCode (opencode)

Other Harbor agents are refused because they cannot verifiably run closed book. Reasoning effort is a first-class run setting next to the agent and model. It is checked per agent before launch, recorded with every job, and kept apart in reports.

Installation

uv add trialdesignbench           # dataset, build, grade, report
uv add "trialdesignbench[judge]"  # + Anthropic SDK for the Anthropic rubric judge
uv add "trialdesignbench[harbor]" # + Harbor to run agents

For development:

git clone https://github.com/BBSW-org/TrialDesignBench.git
cd TrialDesignBench
uv sync --dev

Quick start

Canonical dataset from curated submissions and protocol/SAP Markdown:

uv run tdb dataset import \
	data/json/*.json \
	--out tmp/dataset \
	--documents docs/
uv run tdb dataset check tmp/dataset

Shared environment image, then Harbor tasks:

uv run tdb env build
uv run tdb build \
	tmp/dataset \
	--out tmp/tasks

Run an agent with the network canary, then aggregate. The agent reads its provider's API key (here the ANTHROPIC_API_KEY). The rubric judge needs the key of its backend, ANTHROPIC_API_KEY by default.

export ANTHROPIC_API_KEY=...
uv run tdb run \
	--tasks tmp/tasks \
	--agent claude-code \
	--model anthropic/claude-opus-5-5 \
	--effort high \
	--n-attempts 3 \
	--canary
uv run tdb report \
	"jobs/<job-name>" \
	--format md

Grade any directory with output.json and output.R, no Harbor needed:

uv run tdb grade \
	path/to/submission \
	--rubrics "tmp/dataset/<task_id>/rubrics.json" \
	--out "graded/<task_id>" \
	--trajectory path/to/trajectory.json

See the documentation for the dataset format, supported agents and their credentials, the judge, environment and network policy, grading rules, and reproducibility guarantees.

Design principles

  • Hidden rubrics. Agents see the prompt template, the question skeleton, and the source document. Rubrics live only in each task's tests/ directory, which runs in a separate verifier container.
  • Decoupled grading. tdb grade is a pure function of the submission (output.json, output.R, trajectory), the rubrics, and the judge configuration. It runs the same inside Harbor, standalone, and in tests, so recorded runs can be regraded and external submissions compared.
  • Fail loudly. Every grading error is an explicit status and zeroes the reward; a missing R runtime or trajectory is never a pass.
  • Closed book is enforced. While the agent runs, egress is an allowlist of model API hosts; server-side web tools are disabled in the harness, the grader scans the trajectory for violations, and a network canary proves the policy holds.
  • One pinned image for agent and verifier: R with a dated CRAN snapshot, Python, Node, pinned agent CLIs and closed book agent settings, and the pharma skills.

Metadata

Release files for trialdesignbench 1.4.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for trialdesignbench 1.4.0
File Size Uploaded Provenance Actions
trialdesignbench-1.4.0.tar.gz 65.1 kB Without provenance Details for trialdesignbench-1.4.0.tar.gz

Built distribution (wheel)

Table of built distributions (wheels) for trialdesignbench 1.4.0
File Interpreter ABI Platform Actions
trialdesignbench-1.4.0-py3-none-any.whl Python 3 none any Details for trialdesignbench-1.4.0-py3-none-any.whl

Total release size: 144.6 kB

Release files / trialdesignbench-1.4.0.tar.gz

Download URL trialdesignbench-1.4.0.tar.gz
Size 65.1 kB
Tags Source
SHA-256 checksum
How to use checksums
d8ae88a0e30d0579e0445d78ea590e652451ed8642063abc2c2cb4793d18676d
BLAKE2b-256 checksum
How to use checksums
8e09c6643ce99d43d0b34e09e6f62b6e59258049c05efb07fc9a706e9ef81681
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via uv/0.12.21 {"installer":{"name":"uv","version":"0.12.21","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

Release files / trialdesignbench-1.4.0-py3-none-any.whl

Download URL trialdesignbench-1.4.0-py3-none-any.whl
Size 79.5 kB
Tags Python 3
SHA-256 checksum
How to use checksums
0f9c638a1e52bf52cf3a7275cfe86cfa0a65194cbc9242269f8fd77b8299f787
BLAKE2b-256 checksum
How to use checksums
b6a60d8d3a78304432203042e1a42968af3f27d86112af43589989b8acf46963
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via uv/0.12.21 {"installer":{"name":"uv","version":"0.12.21","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

Security for trialdesignbench 1.4.0

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page