chore(style): reduce line length and truncate compression

Signed-off-by: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com>

openllm-python/tests/_strategies/_configuration.py

@@ -24,7 +24,12 @@ def model_settings(draw: st.DrawFn):
       "workers_per_resource": st.one_of(st.integers(min_value=1), st.floats(min_value=0.1, max_value=1.0)),
   }
   return draw(st.builds(ModelSettings, **kwargs))
-def make_llm_config(cls_name: str, dunder_config: dict[str, t.Any] | ModelSettings, fields: tuple[tuple[t.LiteralString, str, t.Any], ...] | None = None, generation_fields: tuple[tuple[t.LiteralString, t.Any], ...] | None = None,) -> type[openllm.LLMConfig]:
+def make_llm_config(
+    cls_name: str,
+    dunder_config: dict[str, t.Any] | ModelSettings,
+    fields: tuple[tuple[t.LiteralString, str, t.Any], ...] | None = None,
+    generation_fields: tuple[tuple[t.LiteralString, t.Any], ...] | None = None,
+) -> type[openllm.LLMConfig]:
   globs: dict[str, t.Any] = {"openllm": openllm}
   _config_args: list[str] = []
   lines: list[str] = [f"class {cls_name}Config(openllm.LLMConfig):"]

openllm-python/tests/configuration_test.py

@@ -14,7 +14,11 @@ def test_missing_default():
   with pytest.raises(ValueError, match="Missing required fields *"):
     make_llm_config("MissingArchitecture", {"default_id": "huggingface/t5-tiny-testing", "model_ids": ["huggingface/t5-tiny-testing"], "requirements": ["bentoml"],},)
 def test_forbidden_access():
-  cl_ = make_llm_config("ForbiddenAccess", {"default_id": "huggingface/t5-tiny-testing", "model_ids": ["huggingface/t5-tiny-testing", "bentoml/t5-tiny-testing"], "architecture": "PreTrainedModel", "requirements": ["bentoml"],},)
+  cl_ = make_llm_config(
+      "ForbiddenAccess", {
+          "default_id": "huggingface/t5-tiny-testing", "model_ids": ["huggingface/t5-tiny-testing", "bentoml/t5-tiny-testing"], "architecture": "PreTrainedModel", "requirements": ["bentoml"],
+      },
+  )
 
   assert pytest.raises(openllm.exceptions.ForbiddenAttributeError, cl_.__getattribute__, cl_(), "__config__",)
   assert pytest.raises(openllm.exceptions.ForbiddenAttributeError, cl_.__getattribute__, cl_(), "GenerationConfig",)
@@ -94,7 +98,11 @@ def test_struct_envvar_with_overwrite_provided_env(monkeypatch: pytest.MonkeyPat
   with monkeypatch.context() as mk:
     mk.setenv(field_env_key("overwrite_with_env_available", "field1"), str(4.0))
     mk.setenv(field_env_key("overwrite_with_env_available", "temperature", suffix="generation"), str(0.2))
-    sent = make_llm_config("OverwriteWithEnvAvailable", {"default_id": "asdfasdf", "model_ids": ["asdf", "asdfasdfads"], "architecture": "PreTrainedModel"}, fields=(("field1", "float", 3.0),),).model_construct_env(field1=20.0, temperature=0.4)
+    sent = make_llm_config(
+        "OverwriteWithEnvAvailable", {
+            "default_id": "asdfasdf", "model_ids": ["asdf", "asdfasdfads"], "architecture": "PreTrainedModel"
+        }, fields=(("field1", "float", 3.0),),
+    ).model_construct_env(field1=20.0, temperature=0.4)
     assert sent.generation_config.temperature == 0.4
     assert sent.field1 == 20.0
 @given(model_settings())

openllm-python/tests/models/conftest.py

@@ -109,7 +109,9 @@ class DockerHandle(_Handle):
     container = self.docker_client.containers.get(self.container_name)
     return container.status in ["running", "created"]
 @contextlib.contextmanager
-def _local_handle(model: str, model_id: str, image_tag: str, deployment_mode: t.Literal["container", "local"], quantize: t.Literal["int8", "int4", "gptq"] | None = None, *, _serve_grpc: bool = False,):
+def _local_handle(
+    model: str, model_id: str, image_tag: str, deployment_mode: t.Literal["container", "local"], quantize: t.Literal["int8", "int4", "gptq"] | None = None, *, _serve_grpc: bool = False,
+):
   with openllm.utils.reserve_free_port() as port:
     pass
 
@@ -129,7 +131,9 @@ def _local_handle(model: str, model_id: str, image_tag: str, deployment_mode: t.
   if proc.stderr:
     proc.stderr.close()
 @contextlib.contextmanager
-def _container_handle(model: str, model_id: str, image_tag: str, deployment_mode: t.Literal["container", "local"], quantize: t.Literal["int8", "int4", "gptq"] | None = None, *, _serve_grpc: bool = False,):
+def _container_handle(
+    model: str, model_id: str, image_tag: str, deployment_mode: t.Literal["container", "local"], quantize: t.Literal["int8", "int4", "gptq"] | None = None, *, _serve_grpc: bool = False,
+):
   envvar = openllm.utils.EnvVarMixin(model)
 
   with openllm.utils.reserve_free_port() as port, openllm.utils.reserve_free_port() as prom_port:
@@ -154,7 +158,11 @@ def _container_handle(model: str, model_id: str, image_tag: str, deployment_mode
   gpus = openllm.utils.device_count() or -1
   devs = [docker.types.DeviceRequest(count=gpus, capabilities=[["gpu"]])] if gpus > 0 else None
 
-  container = client.containers.run(image_tag, command=args, name=container_name, environment=env, auto_remove=False, detach=True, device_requests=devs, ports={"3000/tcp": port, "3001/tcp": prom_port},)
+  container = client.containers.run(
+      image_tag, command=args, name=container_name, environment=env, auto_remove=False, detach=True, device_requests=devs, ports={
+          "3000/tcp": port, "3001/tcp": prom_port
+      },
+  )
 
   yield DockerHandle(client, container.name, port, deployment_mode)
 

openllm-python/tests/package_test.py

@@ -5,7 +5,9 @@ if t.TYPE_CHECKING: from pathlib import Path
 
 HF_INTERNAL_T5_TESTING = "hf-internal-testing/tiny-random-t5"
 
-actions_xfail = functools.partial(pytest.mark.xfail, condition=os.getenv("GITHUB_ACTIONS") is not None, reason="Marking GitHub Actions to xfail due to flakiness and building environment not isolated.",)
+actions_xfail = functools.partial(
+    pytest.mark.xfail, condition=os.getenv("GITHUB_ACTIONS") is not None, reason="Marking GitHub Actions to xfail due to flakiness and building environment not isolated.",
+)
 @actions_xfail
 def test_general_build_with_internal_testing():
   bento_store = BentoMLContainer.bento_store.get()