Fix linting issues: isort, ruff, and trailing whitespace

Signed-off-by: Xing Liu <xingliu14@gmail.com>

Author: xingliu14

tpu_inference/__init__.py

@@ -2,7 +2,8 @@
 # modules to ensure that the environment variables are set before any
 # other modules are imported.
 import tpu_inference.env_override  # noqa: F401
-from tpu_inference import envs, tpu_info as ti
+from tpu_inference import envs
+from tpu_inference import tpu_info as ti
 from tpu_inference.logger import init_logger
 
 logger = init_logger(__name__)

tpu_inference/mock/vllm_envs.py

@@ -191,7 +191,7 @@ def maybe_convert_bool(value: Optional[str]) -> Optional[bool]:
 
 def _get_jax_platforms() -> str:
     """Get JAX_PLATFORMS from tpu_inference.envs module.
-    
+
     Returns:
         The JAX_PLATFORMS value.
     """

tpu_inference/utils.py

@@ -1,5 +1,4 @@
 # SPDX-License-Identifier: Apache-2.0
-import os
 import time
 from collections import defaultdict
 from collections.abc import Sequence
@@ -14,7 +13,8 @@
 from jax._src import xla_bridge as xb
 from jax._src.lib import xla_client as xc
 from jax.sharding import Mesh, NamedSharding, PartitionSpec
-from vllm import envs as vllm_envs, utils
+from vllm import envs as vllm_envs
+from vllm import utils
 
 from tpu_inference import envs
 from tpu_inference.logger import init_logger