minor re-formatting

Author: JamesDeng42

src/maxtext/eval/runner/harness_runner.py

@@ -159,14 +159,14 @@ def run_harness(cfg: dict, hf_token: str | None = None) -> dict:
           lm_model_type,
           server.base_url,
       )
-      simple_eval_kwargs: dict = dict(
-          model=lm_model_type,
-          model_args=model_args,
-          tasks=tasks,
-          num_fewshot=num_fewshot,
-          limit=num_samples,
-          log_samples=False,
-      )
+      simple_eval_kwargs: dict = {
+          "model": lm_model_type,
+          "model_args": model_args,
+          "tasks": tasks,
+          "num_fewshot": num_fewshot,
+          "limit": num_samples,
+          "log_samples": False,
+      }
       if apply_chat_template:
         simple_eval_kwargs["apply_chat_template"] = True
       if fewshot_as_multiturn:

src/maxtext/eval/runner/server_manager.py

@@ -18,7 +18,6 @@
 
 import logging
 import os
-import sys
 import threading
 import time
 import uuid

tests/unit/eval/test_build_app.py

@@ -242,8 +242,6 @@ def test_completions_token_id_prompt_normalised_as_single(self):
 
   def test_completions_top_logprobs_populated(self):
     """When logprobs is requested, top_logprobs entries must be non-None dicts."""
-    from types import SimpleNamespace  # pylint: disable=import-outside-toplevel
-
     mock_output = _make_mock_output(generated_text="hi", prompt_token_ids=[1], generated_token_ids=[4, 5])
     mock_output.outputs[0].logprobs = [
         {4: SimpleNamespace(logprob=-0.5), 6: SimpleNamespace(logprob=-1.0)},