File tree Expand file tree Collapse file tree 1 file changed +4
-5
lines changed Expand file tree Collapse file tree 1 file changed +4
-5
lines changed Original file line number Diff line number Diff line change @@ -353,7 +353,8 @@ def read_data_with_scan(self,
353
353
temp_query = query .copy ()
354
354
query .clear ()
355
355
query ["query" ] = temp_query
356
- pr_bar : tqdm .tqdm = None
356
+ pr_bar = tqdm .tqdm (desc = "CogStack retrieved..." ,
357
+ disable = not show_progress , colour = 'green' )
357
358
358
359
scan_results = es_helpers .scan (self .elastic ,
359
360
index = index ,
@@ -364,10 +365,8 @@ def read_data_with_scan(self,
364
365
fields = include_fields ,
365
366
allow_no_indices = False ,)
366
367
all_mapped_results = []
367
- results = self .elastic .count (index = index , query = query ["query" ])
368
- pr_bar = tqdm .tqdm (scan_results , total = results ["count" ],
369
- desc = "CogStack retrieved..." ,
370
- disable = not show_progress , colour = 'green' )
368
+ pr_bar .iterable = scan_results
369
+ pr_bar .total = self .elastic .count (index = index , query = query ["query" ])["count" ]
371
370
all_mapped_results = self .__map_search_results (hits = pr_bar )
372
371
except BaseException as err :
373
372
if isinstance (err , KeyboardInterrupt ):
You can’t perform that action at this time.
0 commit comments