[Feat] Support Pipeline Resolution Based on HF Config & Make Pipeline Registry Eager (#3760)

Signed-off-by: Alex Brooks <albrooks@redhat.com>

tests/utils/test_tracking_parser.py

@@ -8,11 +8,11 @@ import pytest
 import yaml
 from vllm.utils.argparse_utils import FlexibleArgumentParser
 
+from vllm_omni.config.config_factory import StageConfigFactory
+from vllm_omni.config.pipeline_registry import OMNI_PIPELINES
 from vllm_omni.config.stage_config import (
-    _PIPELINE_REGISTRY,
     DeployConfig,
     PipelineConfig,
-    StageConfigFactory,
     StageDeployConfig,
     StagePipelineConfig,
 )
@@ -44,9 +44,9 @@ _TEST_DEPLOY = DeployConfig(
 @pytest.fixture()
 def mock_stages(monkeypatch):
     """Register a fake pipeline and mock deploy YAML loading."""
-    monkeypatch.setitem(_PIPELINE_REGISTRY._loaded, _TEST_MODEL, _TEST_PIPELINE)
+    monkeypatch.setitem(OMNI_PIPELINES, _TEST_MODEL, _TEST_PIPELINE)
     monkeypatch.setattr(
-        "vllm_omni.config.stage_config.load_deploy_config",
+        "vllm_omni.config.config_factory.load_deploy_config",
         lambda _path: _TEST_DEPLOY,
     )
     return __file__
@@ -462,6 +462,7 @@ def test_explicit_cli_arg_reaches_runtime_overrides(mock_stages):
     explicit_kwargs = ns.get_explicit_kwargs_dict()
     stages = StageConfigFactory._create_from_registry(
         _TEST_MODEL,
+        _TEST_PIPELINE,
         explicit_kwargs,
         deploy_config_path=mock_stages,
     )
@@ -478,6 +479,7 @@ def test_omitted_default_not_in_runtime_overrides(mock_stages):
     explicit_kwargs = ns.get_explicit_kwargs_dict()
     stages = StageConfigFactory._create_from_registry(
         _TEST_MODEL,
+        _TEST_PIPELINE,
         explicit_kwargs,
         deploy_config_path=mock_stages,
     )
@@ -499,6 +501,7 @@ def test_config_file_args_reach_runtime_overrides(mock_stages):
     explicit_kwargs = ns.get_explicit_kwargs_dict()
     stages = StageConfigFactory._create_from_registry(
         _TEST_MODEL,
+        _TEST_PIPELINE,
         explicit_kwargs,
         deploy_config_path=mock_stages,
     )
@@ -516,6 +519,7 @@ def test_per_stage_override_routes_correctly(mock_stages):
     explicit_kwargs = ns.get_explicit_kwargs_dict()
     stages = StageConfigFactory._create_from_registry(
         _TEST_MODEL,
+        _TEST_PIPELINE,
         explicit_kwargs,
         deploy_config_path=mock_stages,
     )
@@ -537,6 +541,7 @@ def test_explicit_args_omitted_from_yaml(mock_stages):
     explicit_kwargs = ns.get_explicit_kwargs_dict()
     stages = StageConfigFactory._create_from_registry(
         _TEST_MODEL,
+        _TEST_PIPELINE,
         explicit_kwargs,
         deploy_config_path=mock_stages,
     )