pass all pre-commit

research_town/agent_base.py

@@ -1,13 +1,15 @@
 import datetime
-import requests
+from typing import Any, Dict, List
 from xml.etree import ElementTree
-from typing import List, Dict, Any
+
+import requests
+
 from .utils import (
-    summarize_research_direction,
+    bfs,
+    generate_ideas,
     get_bert_embedding,
+    summarize_research_direction,
     summarize_research_field,
-    generate_ideas,
-    bfs,
 )
 
 

research_town/env_base.py

@@ -1,4 +1,5 @@
 from typing import Dict
+
 from .agent_base import BaseResearchAgent
 from .kb_base import BaseKnowledgeBase
 

research_town/env_paper_rebuttal.py

@@ -1,4 +1,5 @@
 from typing import Dict
+
 from .env_base import BaseMultiAgentEnv
 
 

research_town/env_paper_submission.py

@@ -1,4 +1,5 @@
 from typing import Dict
+
 from .env_base import BaseMultiAgentEnv
 
 

research_town/kb_base.py

@@ -1,4 +1,5 @@
 from typing import Dict, List
+
 from .utils import get_daily_papers
 
 

research_town/utils.py

@@ -1,15 +1,16 @@
-import os
-import json
 import datetime
-import openai
+import json
+import os
 import time
+from typing import Any, Dict, List, Tuple
+
+import arxiv
 import faiss
-from transformers import BertTokenizer, BertModel
+import openai
 import torch
-import arxiv
-from tqdm import tqdm
-from typing import Dict, List, Tuple, Any
 from arxiv import Client, Search
+from tqdm import tqdm
+from transformers import BertModel, BertTokenizer
 
 KEY = "7a1821d4e4a3e41e3d523e97e0fd8950dedac2824aef99cb19d550500cb21a42"
 openai.api_base = "https://api.together.xyz"