remove unused imports

Author: EYH0602

pyproject.toml

@@ -24,7 +24,7 @@ dependencies = [
     "pytest>=8.0.0",
     "python-dotenv==1.0.1",
     "requests==2.32.3",
-    "returns[compatible-mypy]==0.22.0",
+    "returns[compatible-mypy]>=0.26.0",
     "seaborn==0.13.2",
     "tabulate>=0.9.0",
     "tenacity>=9.1.2",

src/main.py

@@ -1,13 +1,12 @@
 import os
-from typing import Callable
 import json
 import logging
 
 from funcy_chain import Chain
 from funcy import lmap
 from tqdm import tqdm
 from dacite import from_dict
-from returns.result import ResultE, Success, Failure
+from returns.result import ResultE
 
 import fire
 from tfbench.common import (

src/tfbench/__init__.py

@@ -1 +0,0 @@
-from . import env

src/tfbench/env.py

@@ -1,6 +1,4 @@
-import os
-
-from dotenv import load_dotenv, dotenv_values
+from dotenv import dotenv_values
 
 ENV = dotenv_values(".env")
 

src/tfbench/experiment.py

@@ -1,14 +1,3 @@
 """
 Experiment script for OpenAI models
 """
-
-from typing import Callable
-import logging
-
-
-from .lm import get_sys_prompt, MAX_TOKENS
-
-DEEPSEEK_MODELS = [
-    "deepseek-reasoner",
-    "deepseek-chat",
-]

src/tfbench/lm/_anthropic.py

@@ -1,5 +1,4 @@
-from typing import Callable
-from anthropic import Anthropic, InternalServerError, NOT_GIVEN
+from anthropic import Anthropic
 from ..env import ENV
 from ._types import LM, LMAnswer, ReasoningEffort, EFFORT_TOKEN_MAP
 from .settings import MAX_TOKENS

src/tfbench/lm/_google.py

@@ -1,7 +1,6 @@
 """Inference helper for Google Gemini"""
 
-from typing import Literal, Mapping
-from enum import Enum
+from typing import Literal
 
 from google import genai
 from google.genai.types import GenerateContentConfig, ThinkingConfig

src/tfbench/lm/_ollama.py

@@ -36,6 +36,6 @@ def _gen(self, prompt: str) -> LMAnswer:
             think=True,
         )
         return LMAnswer(
-            answer=response.message.content,
-            reasoning_steps=response.message.thinking,
+            answer=response.message.content,  # type: ignore
+            reasoning_steps=response.message.thinking,  # type: ignore
         )

src/tfbench/lm/_openai.py

@@ -1,5 +1,3 @@
-from typing import get_args
-
 from openai import OpenAI, NOT_GIVEN
 from openai.types.shared.reasoning_effort import ReasoningEffort
 

src/tfbench/lm/_types.py

@@ -1,5 +1,4 @@
 from abc import ABC, abstractmethod
-from enum import StrEnum
 from dataclasses import dataclass
 from typing import Literal