fix(bench): minor fix on mlbench (#801)

ulab-uiuc · Oct 23, 2024 · 5c8b327 · 5c8b327
1 parent abe9c0a
commit 5c8b327
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/research_bench/mlbench.py b/research_bench/mlbench.py
@@ -3,14 +3,14 @@
 
 from tqdm import tqdm
 from utils import (
+    get_arxiv_ids_from_keyword,
     get_author_data,
     get_paper_data,
     get_proposal_from_paper,
     save_benchmark,
 )
 
 from research_town.configs import Config
-from research_town.utils.paper_collector import get_paper_by_keyword
 
 
 def process_keywords(
@@ -24,7 +24,7 @@ def process_keywords(
 
     for keyword in keywords:
         print(f"Fetching papers for keyword: '{keyword}'")
-        arxiv_ids = get_paper_by_keyword(
+        arxiv_ids = get_arxiv_ids_from_keyword(
             keyword, existing_arxiv_ids, max_papers_per_keyword
         )
         for arxiv_id in tqdm(arxiv_ids, desc=f"Processing papers for '{keyword}'"):