feat: bind pipeline with runtime manager

This commit is contained in:
Junyan Qin
2025-03-28 15:55:03 +08:00
parent 5379e4cf27
commit 7cd03b0243
7 changed files with 119 additions and 10 deletions
+2 -2
View File
@@ -7,13 +7,13 @@ from ..core import app, entities as core_entities
from . import entities
_stage_classes: dict[str, PipelineStage] = {}
preregistered_stages: dict[str, PipelineStage] = {}
def stage_class(name: str):
def decorator(cls):
_stage_classes[name] = cls
preregistered_stages[name] = cls
return cls
return decorator