Skip to content

Minor: enable MD progress bar#448

Open
danielzuegner wants to merge 3 commits intoTorchSim:mainfrom
danielzuegner:integration-progress-bar
Open

Minor: enable MD progress bar#448
danielzuegner wants to merge 3 commits intoTorchSim:mainfrom
danielzuegner:integration-progress-bar

Conversation

@danielzuegner
Copy link
Contributor

@danielzuegner danielzuegner commented Feb 10, 2026

Summary

Currently, ts.integrate only shows a progress bar when autobatching is enabled, and the progress bar is over the systems completed in the batch. This PR also adds a progress bar to the "inner" MD integration, which I find to be useful especially for longer-running MD. The default is to set leave=False, i.e., the inner progress bar disappears after it is complete, which avoids clogging the log for large batches where we have many inner MD loops.

Checklist

Before a pull request can be merged, the following items must be checked:

  • Doc strings have been added in the Google docstring format.
  • Run ruff on your code.
  • Tests have been added for any new functionality or bug fixes.

Copy link
Collaborator

@orionarcher orionarcher left a comment

Choose a reason for hiding this comment

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

LGTM! Might be nice to include some common/useful kwarg options in the dict for those unfamiliar with pbar.

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.

2 participants