pre-commit formatting

genai_impact/tracers/mistralai_tracer.py

@@ -6,8 +6,8 @@
 from genai_impact.model_repository import models
 
 try:
-    from mistralai.client import MistralClient as _MistralClient
     from mistralai.async_client import MistralAsyncClient as _MistralAsyncClient
+    from mistralai.client import MistralClient as _MistralClient
     from mistralai.models.chat_completion import (
         ChatCompletionResponse as _ChatCompletionResponse,
     )
@@ -46,7 +46,7 @@ async def mistralai_async_chat_wrapper(
     wrapped: Callable,
     instance: _MistralAsyncClient,
     args: Any,
-    kwargs: Any,  # noqa: ARG001
+    kwargs: Any,
 ) -> ChatCompletionResponse:
     response = await wrapped(*args, **kwargs)
     return compute_impacts_and_return_response(response)

genai_impact/tracers/openai_tracer.py

@@ -1,7 +1,6 @@
 from typing import Any, Callable
 
-from openai.resources.chat import Completions, AsyncCompletions
-
+from openai.resources.chat import AsyncCompletions, Completions
 from openai.types.chat import ChatCompletion as _ChatCompletion
 from wrapt import wrap_function_wrapper
 
@@ -36,7 +35,7 @@ async def openai_async_chat_wrapper(
     wrapped: Callable,
     instance: AsyncCompletions,
     args: Any,
-    kwargs: Any,  # noqa: ARG001
+    kwargs: Any,
 ) -> ChatCompletion:
     response = await wrapped(*args, **kwargs)
     return compute_impacts_and_return_response(response)