File tree 1 file changed +7
-10
lines changed
1 file changed +7
-10
lines changed Original file line number Diff line number Diff line change 28
28
29
29
from .futures import MessageFuture
30
30
31
- try :
32
- from tqdm import tqdm
33
- use_progressbar = True
34
- except ImportError :
35
- use_progressbar = False
36
-
37
-
38
31
def _raw_text (s ):
39
32
display_pretty (s , raw = True )
40
33
@@ -498,14 +491,18 @@ def wall_time(self):
498
491
"""
499
492
return self .timedelta (self .submitted , self .received )
500
493
501
- def wait_interactive (self , interval = 1. , timeout = - 1 ):
502
- """interactive wait, printing progress at regular intervals"""
494
+ def wait_interactive (self , interval = 1. , timeout = - 1 , progress = None ):
495
+ """interactive wait, printing progress at regular intervals.
496
+
497
+ progress can be a tqdm-like progress bar."""
498
+
499
+ use_progressbar = progress is not None
503
500
if timeout is None :
504
501
timeout = - 1
505
502
N = len (self )
506
503
tic = time .time ()
507
504
if use_progressbar :
508
- progress_bar = tqdm (total = N )
505
+ progress_bar = progress (total = N )
509
506
n_prev = 0
510
507
while not self .ready () and (timeout < 0 or time .time () - tic <= timeout ):
511
508
self .wait (interval )
You can’t perform that action at this time.
0 commit comments