fix: pre-commit

ecologits/impacts/llm.py

@@ -3,7 +3,7 @@
 from typing import Any, Optional
 
 from ecologits.impacts.dag import DAG
-from ecologits.impacts.modeling import GWP, PE, ADPe, Embodied, Energy, Impacts, Usage, ValueOrRange
+from ecologits.impacts.modeling import GWP, PE, ADPe, Embodied, Energy, Impacts, Usage
 from ecologits.range_value import RangeValue, ValueOrRange
 
 MODEL_QUANTIZATION_BITS = 4

ecologits/model_repository.py

@@ -59,7 +59,7 @@ class Models(BaseModel):
 
 class ModelRepository:
 
-    def __init__(self, models: list[Model], aliases: Optional[list[Alias]] = None):
+    def __init__(self, models: list[Model], aliases: Optional[list[Alias]] = None) -> None:
         self.__models: dict[tuple[str, str], Model] = {}
         for m in models:
             key = m.provider.value, m.name
@@ -89,7 +89,7 @@ def from_json(cls, filepath: Optional[str] = None) -> "ModelRepository":
             filepath = os.path.join(
                 os.path.dirname(os.path.realpath(__file__)), "data", "models.json"
             )
-        with open(filepath, "r") as fd:
+        with open(filepath) as fd:
             data = json.load(fd)
             mf = Models.model_validate(data)
         return cls(models=mf.models, aliases=mf.aliases)

ecologits/range_value.py

@@ -1,7 +1,7 @@
 from typing import Any, Union
-from typing_extensions import Self
 
 from pydantic import BaseModel, model_validator
+from typing_extensions import Self
 
 
 class RangeValue(BaseModel):

ecologits/tracers/litellm_tracer.py

@@ -1,5 +1,5 @@
 import time
-from typing import Any, Callable, Union, Optional
+from typing import Any, Callable, Optional, Union
 
 from wrapt import wrap_function_wrapper
 
@@ -13,7 +13,7 @@
     from litellm import AsyncCompletions, Completions
     from litellm.types.utils import ModelResponse
     from litellm.utils import CustomStreamWrapper
-    from rapidfuzz import process, fuzz
+    from rapidfuzz import fuzz, process
 
 except ImportError:
     ModelResponse = object()
@@ -35,7 +35,7 @@ class ChatCompletionChunk(ModelResponse):
 _model_choices = [f"{m.provider.value}/{m.name}" for m in models.list_models()]
 
 
-def litellm_match_model(model_name) -> Optional[tuple[str, str]]:
+def litellm_match_model(model_name: str) -> Optional[tuple[str, str]]:
     """
     Match according provider and model from a litellm model_name.
 
@@ -45,7 +45,6 @@ def litellm_match_model(model_name) -> Optional[tuple[str, str]]:
     Returns:
         A tuple (provider, model_name) matching a record of the ModelRepository.
     """
-    # print(process.extractOne(query=model_name, choices=_model_choices, scorer=fuzz.token_sort_ratio))
     candidate = process.extractOne(
         query=model_name,
         choices=_model_choices,
@@ -55,6 +54,7 @@ def litellm_match_model(model_name) -> Optional[tuple[str, str]]:
     if candidate is not None:
         provider, model_name = candidate[0].split("/", 1)
         return provider, model_name
+    return None
 
 
 def litellm_chat_wrapper(

ecologits/tracers/utils.py

@@ -4,8 +4,7 @@
 from ecologits.impacts.llm import compute_llm_impacts
 from ecologits.impacts.modeling import Impacts
 from ecologits.log import logger
-from ecologits.impacts.modeling import Impacts
-from ecologits.model_repository import models, ArchitectureTypes
+from ecologits.model_repository import ArchitectureTypes, models
 
 
 def _avg(value_range: tuple) -> float: