fix: Formatting/typing issues

Author: bradhilton

src/art/client.py

@@ -2,6 +2,7 @@
 from typing import Any, Iterable, Literal, TypedDict, cast
 
 import httpx
+from openai import AsyncOpenAI, BaseModel, _exceptions
 from openai._base_client import AsyncAPIClient, AsyncPaginator, make_request_options
 from openai._compat import cached_property
 from openai._qs import Querystring
@@ -14,8 +15,6 @@
 from openai.resources.models import AsyncModels  # noqa: F401
 from typing_extensions import override
 
-from openai import AsyncOpenAI, BaseModel, _exceptions
-
 from .trajectories import TrajectoryGroup
 
 

src/art/preprocessing/tokenize.py

@@ -213,15 +213,15 @@ def tokenize_trajectory(
                     for token_logprob in token_logprobs
                 )
             except (IndexError, ValueError):
-                token_ids[start:end] = [
+                token_ids[start:end] = [  # type: ignore
                     token_id if token_id is not None else tokenizer.eos_token_id
                     for token_id in tokenizer.convert_tokens_to_ids(
                         [
                             token_logprob.token or tokenizer.eos_token
                             for token_logprob in token_logprobs
                         ]
-                    )
-                ]  # type: ignore
+                    )  # type: ignore
+                ]
             logprobs[start:end] = (
                 token_logprob.logprob for token_logprob in token_logprobs
             )

src/art/serverless/backend.py

@@ -104,7 +104,7 @@ async def _log(
         self,
         model: "Model",
         trajectory_groups: list[TrajectoryGroup],
-        split: str,
+        split: str = "val",
     ) -> None:
         # TODO: log trajectories to local file system?
         if not model.trainable: