5c14757ef0
Signed-off-by: knlnguyen1802 <knlnguyen1802@gmail.com> Signed-off-by: Samit <285365963@qq.com> Co-authored-by: Samit <285365963@qq.com> Co-authored-by: SYLAR <125541396+lishunyang12@users.noreply.github.com>
22 lines
754 B
Python
22 lines
754 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
"""Test-only worker extension for custom diffusion pipeline E2E tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from vllm_omni.diffusion.worker.diffusion_worker import CustomPipelineWorkerExtension
|
|
|
|
|
|
class vLLMOmniColocateWorkerExtensionForTest(CustomPipelineWorkerExtension):
|
|
"""Minimal worker extension used by tests.
|
|
|
|
This intentionally stays lightweight: we only inherit the base custom
|
|
pipeline extension and add one test function.
|
|
"""
|
|
|
|
@staticmethod
|
|
def test_extension_name() -> str:
|
|
"""Return a stable identifier for assertions in unit tests."""
|
|
return "vllm-omni-colocate-worker-extension-for-test"
|