nits

Signed-off-by: rickyx <rickyx@anyscale.com>

Author: rickyyx

vllm/engine/arg_utils.py

@@ -1214,7 +1214,7 @@ def create_engine_config(self,
 
 
 @dataclass
-class V1EngineArgs(EngineArgs):
+class EngineArgsV1(EngineArgs):
     """Arguments for vLLM engine v1."""
 
     # V1's default values that differ from the default values in EngineArgs.
@@ -1232,7 +1232,7 @@ def create_engine_config(self,
                              usage_context: Optional[UsageContext] = None
                              ) -> VllmConfig:
         assert (usage_context
-                is not None), "usage_context must be provided for V1EngineArgs"
+                is not None), "usage_context must be provided for EngineArgsV1"
 
         if self.max_num_batched_tokens is None:
             if usage_context == UsageContext.LLM_CLASS:
@@ -1256,9 +1256,9 @@ def create_engine_config(self,
 
 
 if envs.VLLM_USE_V1:
-    # Overwrite EngineArgs to use V1EngineArgs
+    # Overwrite EngineArgs to use EngineArgsV1
     # This has to be done before `AsyncEngineArgs` is imported.
-    EngineArgs = V1EngineArgs  # type: ignore
+    EngineArgs = EngineArgsV1  # type: ignore
 
 
 @dataclass