Skip to content

feat: add --dry-run to aidata download dataset command - #55

Draft
danellecline with Copilot wants to merge 2 commits into
mainfrom
copilot/add-dry-run-to-aidata-download
Draft

feat: add --dry-run to aidata download dataset command#55
danellecline with Copilot wants to merge 2 commits into
mainfrom
copilot/add-dry-run-to-aidata-download

Conversation

Copilot AI commented Jun 9, 2026

Copy link
Copy Markdown
Contributor

The aidata download dataset command lacked a --dry-run option, making it impossible to preview what would be fetched without triggering actual file writes and media downloads.

Changes

  • commands/download.py: Adds @common_args.dry_run decorator and dry_run: bool parameter; skips output directory creation and passes the flag through to the generator.
  • generators/coco_voc.py: Adds dry_run: bool = False parameter; after querying and counting records from the Tator API, returns early with a log summary when set — no files written, no media downloaded.
  • tests/test_download.py (new): Adds test_download_dataset_dry_run_flag_in_help (no-database) and a database-gated integration test that asserts no images/ or labels/ directories are created.

Example

aidata download dataset \
  --dry-run \
  --config config.yml \
  --token $TATOR_TOKEN \
  --version Baseline
# Logs: "Dry run - would download 1234 records for version ['Baseline'] ..."
# No files written.

Copilot AI changed the title [WIP] Add --dry-run option to aidata download dataset command feat: add --dry-run to aidata download dataset command Jun 9, 2026
Copilot AI requested a review from danellecline June 9, 2026 21:04
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add --dry-run to aidata download dataset command

2 participants