feat(engine): add rid parameter to methods in Engine class (#13095)

This commit is contained in:
ishandhanani
2025-11-11 22:37:54 -08:00
committed by GitHub
parent 7ea5b42d70
commit 1a5c313f97
2 changed files with 9 additions and 0 deletions

View File

@@ -29,6 +29,7 @@ class EngineBase(ABC):
bootstrap_port: Optional[Union[List[int], int]] = None,
bootstrap_room: Optional[Union[List[int], int]] = None,
data_parallel_rank: Optional[int] = None,
rid: Optional[Union[List[str], str]] = None,
) -> Union[Dict, Iterator[Dict]]:
"""Generate outputs based on given inputs."""
pass