Skip to content

Commit

Permalink
feat: add progress hook to get_collection function for improved tracking
Browse files Browse the repository at this point in the history
  • Loading branch information
Ovler-Young committed Nov 19, 2024
1 parent 886cdf7 commit b367a5b
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/ia_collection_analyzer/libs.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,8 @@ def get_collection(collection_id, progress_hook=None) -> list:
)
collection = []
total_items = search.num_found
if progress_hook:
progress_hook(0, total_items)
for result in tqdm(
search, desc=f"Fetching {collection_id}", total=search.num_found
):
Expand Down

0 comments on commit b367a5b

Please sign in to comment.