lint

Author: nithinvc

skyrl/backends/skyrl_train/inference_engines/ray_wrapped_inference_engine.py

@@ -136,7 +136,6 @@ def create_ray_wrapped_inference_engines(
 
     if backend == "vllm":
         import vllm
-
         from skyrl.backends.skyrl_train.inference_engines.vllm.vllm_engine import (
             AsyncVLLMRayActor,
             VLLMRayActor,

skyrl/backends/skyrl_train/inference_engines/vllm/vllm_engine.py

@@ -13,8 +13,23 @@
 from uuid import uuid4
 
 import ray
-import vllm
 from loguru import logger
+
+import vllm
+from skyrl.backends.skyrl_train.inference_engines.base import (
+    InferenceEngineInput,
+    InferenceEngineInterface,
+    InferenceEngineOutput,
+)
+from skyrl.backends.skyrl_train.inference_engines.vllm.utils import pop_openai_kwargs
+
+# Backward compatibility: WorkerWrap has moved to inference_servers.vllm_worker
+# This alias preserves the old import path for existing scripts/configs.
+# TODO (Kourosh): Remove this alias once all references are updated.
+from skyrl.backends.skyrl_train.inference_servers.vllm_worker import (
+    WorkerWrap,  # noqa: F401, E402
+)
+from skyrl.backends.skyrl_train.weight_sync import WeightLoader, WeightUpdateRequest
 from vllm import SamplingParams
 from vllm.entrypoints.openai.chat_completion.protocol import (
     ChatCompletionRequest,
@@ -36,21 +51,6 @@
 from vllm.inputs import TokensPrompt
 from vllm.lora.request import LoRARequest
 
-from skyrl.backends.skyrl_train.inference_engines.base import (
-    InferenceEngineInput,
-    InferenceEngineInterface,
-    InferenceEngineOutput,
-)
-from skyrl.backends.skyrl_train.inference_engines.vllm.utils import pop_openai_kwargs
-
-# Backward compatibility: WorkerWrap has moved to inference_servers.vllm_worker
-# This alias preserves the old import path for existing scripts/configs.
-# TODO (Kourosh): Remove this alias once all references are updated.
-from skyrl.backends.skyrl_train.inference_servers.vllm_worker import (
-    WorkerWrap,  # noqa: F401, E402
-)
-from skyrl.backends.skyrl_train.weight_sync import WeightLoader, WeightUpdateRequest
-
 
 @dataclass
 class Logprob:

skyrl/backends/skyrl_train/inference_engines/vllm/vllm_server.py

@@ -2,8 +2,9 @@
 import signal
 
 import uvloop
-import vllm.envs as envs
 from fastapi import Request
+
+import vllm.envs as envs
 from vllm import AsyncLLMEngine
 from vllm.engine.arg_utils import AsyncEngineArgs
 from vllm.entrypoints.launcher import serve_http

skyrl/benchmarks/benchmark_memory.py

@@ -77,9 +77,10 @@
 from typing import Callable, Literal
 
 import httpx
+from transformers import AutoTokenizer
+
 import tinker
 from tinker import types
-from transformers import AutoTokenizer
 
 # Default configuration
 DEFAULT_BASE_MODEL = "Qwen/Qwen3-4B-Instruct-2507"

tests/tinker/test_api.py

@@ -9,13 +9,13 @@
 from urllib.parse import urlparse
 
 import pytest
-import tinker
-from tinker import types
 from transformers import AutoTokenizer
 
+import tinker
 from skyrl.tinker.api import _build_uv_run_cmd_engine
 from skyrl.tinker.config import EngineConfig
 from tests.tinker.conftest import wait_for_condition
+from tinker import types
 
 BASE_MODEL = "trl-internal-testing/tiny-Qwen3ForCausalLM"