Update arvix.py

arvix.py

@@ -55,8 +55,8 @@ def extract_data(category):
     return list(all_ids)
 
 def extract_arxiv_data():
-    # if not utils.download_datafile('arxiv.txt'):
-    #     raise Exception("Failed to download datafile")
+    if not utils.download_datafile('arxiv.txt'):
+        raise Exception("Failed to download datafile")
     categories = {
         "Astrophysics": ["astro-ph"],
         "Condensed Matter": ["cond-mat"],
@@ -93,8 +93,8 @@ def extract_arxiv_data():
         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'):
-    #     raise Exception("Failed to upload datafile")
+    if not utils.upload_datafile('arxiv.txt'):
+        raise Exception("Failed to upload datafile")
     return json.dumps(data, indent=4, ensure_ascii=False)
 
 if __name__ == '__main__':