chore: run pre-commits

genai_impact/__init__.py

@@ -1,7 +1,3 @@
-from genai_impact.client import OpenAI
-from genai_impact.client import MistralClient
+from genai_impact.client import MistralClient, OpenAI
 
-__all__ = [
-    "OpenAI",
-    "MistralClient"
-]
+__all__ = ["OpenAI", "MistralClient"]

genai_impact/client/__init__.py

@@ -1,7 +1,4 @@
-from .openai_wrapper import OpenAI
 from .mistralai_wrapper import MistralClient
+from .openai_wrapper import OpenAI
 
-__all__ = [
-    "OpenAI",
-    "MistralClient"
-]
+__all__ = ["OpenAI", "MistralClient"]

genai_impact/client/mistralai_wrapper.py

@@ -1,22 +1,24 @@
-from typing import Callable, Any
+from typing import Any, Callable
 
 from wrapt import wrap_function_wrapper
 
-from genai_impact.compute_impacts import compute_llm_impact, Impacts
+from genai_impact.compute_impacts import Impacts, compute_llm_impact
 
 try:
     from mistralai.client import MistralClient as _MistralClient
-    from mistralai.models.chat_completion import ChatCompletionResponse as _ChatCompletionResponse
+    from mistralai.models.chat_completion import (
+        ChatCompletionResponse as _ChatCompletionResponse,
+    )
 except ImportError:
     _MistralClient = object()
     _ChatCompletionResponse = object()
 
 
 _MODEL_SIZES = {
     "mistral-tiny": 7.3,
-    "mistral-small": 12.9,      # mixtral active parameters count
+    "mistral-small": 12.9,  # mixtral active parameters count
     "mistral-medium": 70,
-    "mistral-large": 220
+    "mistral-large": 220,
 }
 
 
@@ -25,7 +27,7 @@ class ChatCompletionResponse(_ChatCompletionResponse):
 
 
 def chat_wrapper(
-    wrapped: Callable, instance: _MistralClient, args: Any, kwargs: Any     # noqa: ARG001
+    wrapped: Callable, instance: _MistralClient, args: Any, kwargs: Any  # noqa: ARG001
 ) -> ChatCompletionResponse:
     response = wrapped(*args, **kwargs)
     model_size = _MODEL_SIZES.get(response.model)
@@ -40,6 +42,4 @@ class MistralClient(_MistralClient):
     def __init__(self, **kwargs: Any) -> None:
         super().__init__(**kwargs)
 
-    wrap_function_wrapper(
-        "mistralai.client", "MistralClient.chat", chat_wrapper
-    )
+    wrap_function_wrapper("mistralai.client", "MistralClient.chat", chat_wrapper)

genai_impact/client/openai_wrapper.py

@@ -5,8 +5,7 @@
 from openai.types.chat import ChatCompletion as _ChatCompletion
 from wrapt import wrap_function_wrapper
 
-from genai_impact.compute_impacts import compute_llm_impact, Impacts
-
+from genai_impact.compute_impacts import Impacts, compute_llm_impact
 
 _MODEL_SIZES = {
     "gpt-4-0125-preview": None,
@@ -34,7 +33,7 @@ class ChatCompletion(_ChatCompletion):
 
 
 def chat_wrapper(
-    wrapped: Callable, instance: Completions, args: Any, kwargs: Any    # noqa: ARG001
+    wrapped: Callable, instance: Completions, args: Any, kwargs: Any  # noqa: ARG001
 ) -> ChatCompletion:
     response = wrapped(*args, **kwargs)
     model_size = _MODEL_SIZES.get(response.model)