-
Notifications
You must be signed in to change notification settings - Fork 159
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[FEAT] Enable Progress Bars for PyRunner and RayRunner #1609
Conversation
samster25
commented
Nov 15, 2023
•
edited
Loading
edited
- Enables progress bars when in interactive mode (Notebook or IPython)
- Implementation for Pyrunner, RayRunner[Local] and RayRunner[Remote]
Codecov Report
Additional details and impacted files@@ Coverage Diff @@
## main #1609 +/- ##
==========================================
+ Coverage 84.59% 87.29% +2.69%
==========================================
Files 54 52 -2
Lines 5201 5115 -86
==========================================
+ Hits 4400 4465 +65
+ Misses 801 650 -151
|
af7ff8c
to
b52f6ec
Compare
self.use_ray_tqdm = use_ray_tqdm | ||
self.show_tasks_bar = show_tasks_bar | ||
self.tqdm_mod = tqdm | ||
self.pbars: dict[int, tqdm] = dict() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Initialize as self.pbars = {-1: ...}
? We could also document here that -1
represents the progress for "All tasks".
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
we actually dont want to initialize the the progress bar until we get a task. Otherwise the total is 0 and it's in a weird state.