Fix pytest warning on arxiv lib (#26)

ulab-uiuc · May 11, 2024 · b314f7e · b314f7e
1 parent 2e4f20e
commit b314f7e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/research_town/agents/agent_prompting.py b/research_town/agents/agent_prompting.py
@@ -130,4 +130,4 @@ def summarize_research_direction(personal_info: str) -> List[str]:
             top_p=0,
         )
     content = completion.choices[0].message["content"]
-    return [content]
+    return [content]
diff --git a/research_town/utils/paper_collection.py b/research_town/utils/paper_collection.py
@@ -45,12 +45,14 @@ def neiborhood_search(
 def get_daily_papers(
     topic: str, query: str = "slam", max_results: int = 2
 ) -> Tuple[Dict[str, Dict[str, List[str]]], str]:
-    search_engine = arxiv.Search(
+    client = arxiv.Client()
+    search = arxiv.Search(
         query=query, max_results=max_results, sort_by=arxiv.SortCriterion.SubmittedDate
     )
+    results = client.results(search)
     content: Dict[str, Dict[str, List[str]]] = {}
     newest_day = ""
-    for result in search_engine.results():
+    for result in results:
         paper_title = result.title
         paper_url = result.entry_id
         paper_abstract = result.summary.replace("\n", " ")