Update arvix.py

arvix.py

@@ -50,7 +50,7 @@ def extract_data(category):
         if not utils.check_data_in_file(paper_id, 'arxiv.txt'):
             utils.write_data_to_file(paper_id, 'arxiv.txt')
             all_ids.add(paper_id)
-        if len(all_ids) >= 4:
+        if len(all_ids) >= 2:
             break
     return list(all_ids)
 
@@ -86,11 +86,11 @@ def extract_arxiv_data():
                 if paper_id not in used_ids:
                     category_ids.add(paper_id)
                     used_ids.add(paper_id)
-                if len(category_ids) == 4:
+                if len(category_ids) == 2:
                     break
-            if len(category_ids) == 4:
+            if len(category_ids) == 2:
                 break
-        while len(category_ids) < 4:
+        while len(category_ids) < 2:
             category_ids.add(random.choice(list(used_ids)))
         data[category] = {"ids": list(category_ids), "count": len(category_ids)}
     if not utils.upload_datafile('arxiv.txt'):