bug fix

StampyAI · Aug 9, 2023 · 78ff5eb · 78ff5eb
1 parent 6b024d6
commit 78ff5eb
Showing 1 changed file with 20 additions and 18 deletions.
diff --git a/align_data/sources/arxiv_papers/arxiv_papers.py b/align_data/sources/arxiv_papers/arxiv_papers.py
@@ -56,22 +56,24 @@ def process_entry(self, item) -> None:
         else:
             authors = paper.get("authors") or []
         authors = [str(a).strip() for a in authors]
-        
+
         return self.make_data_entry(
-            url=self.get_item_key(item),
-            source=self.name,
-            source_type=paper["data_source"],
-            title=self.is_val(item.title) or paper.get("title"),
-            authors=authors,
-            date_published=self._get_published_date(
-                self.is_val(item.date_published) or paper.get("date_published")
-            ),
-            data_last_modified=str(metadata.updated),
-            summary=metadata.summary.replace("\n", " "),
-            author_comment=metadata.comment,
-            journal_ref=metadata.journal_ref,
-            doi=metadata.doi,
-            primary_category=metadata.primary_category,
-            categories=metadata.categories,
-            text=paper["text"],
-        )
+            {
+                "url": self.get_item_key(item),
+                "source": self.name,
+                "source_type": paper["data_source"],
+                "title": self.is_val(item.title) or paper.get("title"),
+                "authors": authors,
+                "date_published": self._get_published_date(
+                    self.is_val(item.date_published) or paper.get("date_published")
+                ),
+                "data_last_modified": str(metadata.updated),
+                "summary": metadata.summary.replace("\n", " "),
+                "author_comment": metadata.comment,
+                "journal_ref": metadata.journal_ref,
+                "doi": metadata.doi,
+                "primary_category": metadata.primary_category,
+                "categories": metadata.categories,
+                "text": paper["text"],
+            }
+        )