Skip to content

Commit d04179b

Browse files
committed
Replace consecutive calls logging with per-query result summary
1 parent ba74f05 commit d04179b

1 file changed

Lines changed: 5 additions & 11 deletions

File tree

scripts/1-fetch/arxiv_fetch.py

Lines changed: 5 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -535,7 +535,7 @@ def query_arxiv(args):
535535
break
536536

537537
LOGGER.info(f"Searching for: {search_query}")
538-
consecutive_empty_calls = 0
538+
papers_found_for_query = 0
539539

540540
for start in range(
541541
0,
@@ -585,6 +585,7 @@ def query_arxiv(args):
585585

586586
total_fetched += 1
587587
papers_found_in_batch += 1
588+
papers_found_for_query += 1
588589

589590
LOGGER.info(
590591
f"Found CC licensed paper: {license_info} - "
@@ -599,16 +600,9 @@ def query_arxiv(args):
599600
break
600601

601602
if papers_found_in_batch == 0:
602-
consecutive_empty_calls += 1
603-
if consecutive_empty_calls >= 2:
604-
LOGGER.info(
605-
f"No new papers in 2 consecutive calls for "
606-
f"query: {search_query}. "
607-
f"Moving over to the next query."
608-
)
609-
break
610-
else:
611-
consecutive_empty_calls = 0
603+
break
604+
605+
LOGGER.info(f"Query '{search_query}' completed: {papers_found_for_query} papers found")
612606

613607
# Save results
614608
if args.enable_save:

0 commit comments

Comments
 (0)