Skip to content

feat: add tqdm bar to the dataset segment script#150

Merged
shuheng-liu merged 2 commits into
mainfrom
feat/segment-script-tqdm
Mar 26, 2026
Merged

feat: add tqdm bar to the dataset segment script#150
shuheng-liu merged 2 commits into
mainfrom
feat/segment-script-tqdm

Conversation

@shuheng-liu
Copy link
Copy Markdown
Member

What this does

Add a tqdm progress bar when running src/opentau/scripts/segment_lerobot_dataset.py

How it was tested

Ran src/opentau/scripts/segment_lerobot_dataset.py and observed tqdm bar.

How to checkout & try? (for the reviewer)

Run src/opentau/scripts/segment_lerobot_dataset.py and observe tqdm bar.

Checklist

  • I have added Google-style docstrings to important functions and ensured function parameters are typed.
  • My PR includes policy-related changes.
    • If the above is checked: I have run the GPU pytests (pytest -m "gpu") and regression tests.

Copy link
Copy Markdown
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

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

Pull request overview

This PR adds a progress indicator to the LeRobot dataset segmentation script to make long-running segment jobs easier to monitor.

Changes:

  • Add tqdm integration to show a progress bar while segmenting frames.
  • Pre-compute total frames across all planned segments to drive the progress bar’s total.
  • Update the progress bar once per produced segment based on segment length.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

WilliamYue37
WilliamYue37 previously approved these changes Mar 23, 2026
@shuheng-liu shuheng-liu merged commit bda8da6 into main Mar 26, 2026
5 checks passed
@shuheng-liu shuheng-liu deleted the feat/segment-script-tqdm branch March 26, 2026 04:46
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.

3 participants