[Core] Validate PipelineConfig in __post_init__ (#6829)
Signed-off-by: Alex Brooks <albrooks@redhat.com>
This commit is contained in:
1 parent
7992c2213e
commit
a65e89cbd4
8 files changed
+121
-84
No files matched your search
@@ -1,3 +1,6 @@
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
||||
|
||||
"""Tests for TrackingArgumentParser and related utilities."""
|
||||
|
||||
import argparse
|
||||
@@ -35,7 +38,7 @@ _TEST_PIPELINE = PipelineConfig(
|
||||
stages=(
|
||||
StagePipelineConfig(stage_id=0, model_stage="stage0"),
|
||||
StagePipelineConfig(stage_id=1, model_stage="stage1"),
|
||||
StagePipelineConfig(stage_id=2, model_stage="stage2"),
|
||||
StagePipelineConfig(stage_id=2, model_stage="stage2", final_output=True),
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
Reference in new issue
Block a user