fix pre-commit errors

research_town/agents/agent_base.py

@@ -1,7 +1,7 @@
 from datetime import datetime
-from beartype.typing import Any, Dict, List, Tuple
 
 from beartype import beartype
+from beartype.typing import Any, Dict, List, Tuple
 
 from ..dbs import (
     AgentAgentDiscussionLog,

research_town/dbs/agent_db.py

@@ -1,7 +1,7 @@
 import json
 import uuid
-from beartype.typing import Any, Dict, List, Optional
 
+from beartype.typing import Any, Dict, List, Optional
 from pydantic import BaseModel, Field
 
 

research_town/dbs/env_db.py

@@ -1,7 +1,7 @@
 import json
 import uuid
-from beartype.typing import Any, Dict, List, Optional, Type, TypeVar
 
+from beartype.typing import Any, Dict, List, Optional, Type, TypeVar
 from pydantic import BaseModel, Field
 
 T = TypeVar('T', bound=BaseModel)

research_town/dbs/paper_db.py

@@ -1,7 +1,7 @@
 import json
 import uuid
-from beartype.typing import Any, Dict, List, Optional
 
+from beartype.typing import Any, Dict, List, Optional
 from pydantic import BaseModel, Field
 
 from ..utils.paper_collector import get_daily_papers

research_town/dbs/progress_db.py

@@ -1,7 +1,7 @@
 import json
 import uuid
-from beartype.typing import Any, Dict, List, Optional, Type, TypeVar
 
+from beartype.typing import Any, Dict, List, Optional, Type, TypeVar
 from pydantic import BaseModel, Field
 
 T = TypeVar('T', bound=BaseModel)

research_town/envs/env_paper_rebuttal.py

@@ -1,6 +1,5 @@
-from beartype.typing import Dict, List, Tuple
-
 from beartype import beartype
+from beartype.typing import Dict, List, Tuple
 
 from ..dbs import (
     AgentPaperMetaReviewLog,

research_town/envs/env_paper_submission.py

@@ -1,6 +1,5 @@
-from beartype.typing import Dict, List
-
 from beartype import beartype
+from beartype.typing import Dict, List
 
 from ..agents.agent_base import BaseResearchAgent
 from ..dbs import (

research_town/evaluators/output_format.py

@@ -1,6 +1,5 @@
 
 from beartype.typing import Type, TypeVar
-
 from pydantic import BaseModel, Extra, Field, validator
 
 T = TypeVar('T', bound=BaseModel)

research_town/evaluators/quality_evaluator.py

@@ -1,5 +1,6 @@
 
 import re
+
 from beartype.typing import Any
 
 from ..utils.decorator import parsing_error_exponential_backoff

research_town/utils/agent_collector.py

@@ -1,6 +1,5 @@
-from beartype.typing import Any, Dict, List, Tuple
-
 from arxiv import Client, Search
+from beartype.typing import Any, Dict, List, Tuple
 from tqdm import tqdm
 
 

research_town/utils/agent_prompter.py

@@ -1,6 +1,5 @@
-from beartype.typing import Dict, List, Union
-
 from beartype import beartype
+from beartype.typing import Dict, List, Union
 
 from .model_prompting import model_prompting
 from .paper_collector import get_related_papers

research_town/utils/decorator.py

@@ -1,8 +1,15 @@
 import math
 import time
 from functools import wraps
-from beartype.typing import Any, Callable, List, Optional, TypeVar, cast
 
+from beartype.typing import (
+    Any,
+    Callable,
+    List,
+    Optional,
+    TypeVar,
+    cast,
+)
 from pydantic import BaseModel
 
 INF = float(math.inf)

research_town/utils/eval_prompter.py

@@ -1,6 +1,5 @@
-from beartype.typing import Dict
-
 from beartype import beartype
+from beartype.typing import Dict
 
 from .model_prompting import model_prompting
 

research_town/utils/logging.py

@@ -1,4 +1,5 @@
 import logging
+
 from beartype.typing import Dict, List, Union
 
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')

research_town/utils/model_prompting.py

@@ -1,7 +1,6 @@
-from beartype.typing import List, Optional
-
 import litellm
 from beartype import beartype
+from beartype.typing import List, Optional
 
 from .decorator import api_calling_error_exponential_backoff
 

research_town/utils/paper_collector.py

@@ -1,11 +1,11 @@
 import datetime
-from beartype.typing import Any, Dict, List, Tuple
 from xml.etree import ElementTree
 
 import arxiv
 import faiss
 import requests
 import torch
+from beartype.typing import Any, Dict, List, Tuple
 from transformers import BertModel, BertTokenizer
 
 ATOM_NAMESPACE = "{http://www.w3.org/2005/Atom}"

research_town/utils/serializer.py

@@ -1,6 +1,6 @@
 import importlib
-from beartype.typing import Any, Dict, List, Set, Tuple, Union
 
+from beartype.typing import Any, Dict, List, Set, Tuple, Union
 from pydantic import BaseModel
 
 

research_town/utils/tools.py

@@ -1,6 +1,7 @@
 import datetime
 import json
 import os
+
 from beartype.typing import Any, Dict
 
 

tests/test_agents.py

@@ -166,4 +166,4 @@ def test_discuss(mock_model_prompting: MagicMock) -> None:
     assert response.agent_to_pk is not None
     assert response.agent_from_pk is not None
     assert response.timestep >= 0
-    assert response.pk is not None
+    assert response.pk is not None

tests/test_dbs.py

@@ -1,4 +1,5 @@
-from unittest.mock import patch
+
+from beartype.typing import Any, Dict, Optional
 
 from research_town.dbs.agent_db import AgentProfile, AgentProfileDB
 from research_town.dbs.env_db import (
@@ -14,17 +15,16 @@
     ResearchProgressDB,
 )
 
-from beartype.typing import Optional, Dict, Any
 
 def test_envlogdb_basic()->None:
     db = EnvLogDB()
     review_log = AgentPaperReviewLog(paper_pk="paper1", agent_pk="agent1", review_score=5, review_content="Good paper")
     rebuttal_log = AgentPaperRebuttalLog(paper_pk="paper1", agent_pk="agent1", rebuttal_content="I disagree with the review")
     meta_review_log = AgentPaperMetaReviewLog(paper_pk="paper1", agent_pk="agent1", decision=True, meta_review="Accept")
     discussion_log = AgentAgentDiscussionLog(
-        agent_from_pk="agent1", 
+        agent_from_pk="agent1",
         agent_from_name='Rex Ying',
-        agent_to_pk="agent2", 
+        agent_to_pk="agent2",
         agent_to_name='John Doe',
         message="Let's discuss this paper"
     )
@@ -100,11 +100,11 @@ def test_agentprofiledb_basic()->None:
 
     success = db.delete("non-existing-pk")
     assert not success
-    
+
     conditions: Dict[str, Any] = {"name": "Jane Smith"}
 
     results = db.get(**conditions)
-    
+
     assert len(results) == 1
     assert results[0].name == "Jane Smith"
 
@@ -216,7 +216,7 @@ def test_researchprogressdb_basic()->None:
 
     db.add(idea1)
     db.add(idea2)
-    
+
     new_idea = ResearchIdea(content="Blockchain research proposal")
     db.add(new_idea)
     assert new_idea.dict() in db.data["ResearchIdea"]
@@ -249,4 +249,3 @@ def test_researchprogressdb_basic()->None:
     new_db.load_from_file(file_name)
 
     assert len(new_db.data["ResearchIdea"]) == 2
-

tests/test_eval.py

@@ -1,7 +1,7 @@
-from beartype.typing import Any
 from unittest.mock import MagicMock, patch
 
 import pytest
+from beartype.typing import Any
 
 from research_town.evaluators.quality_evaluator import (
     IdeaQualityEvaluator,