diff --git a/darwin/future/meta/objects/stage.py b/darwin/future/meta/objects/stage.py index 85ae3b0ba..6ca58683e 100644 --- a/darwin/future/meta/objects/stage.py +++ b/darwin/future/meta/objects/stage.py @@ -41,4 +41,25 @@ def move_attached_files_to_stage(self, new_stage_id: UUID) -> Stage: @property def id(self) -> UUID: + """Stage ID.""" return self._element.id + + @property + def name(self) -> str: + """Stage name.""" + return self._element.name + + @property + def type(self) -> str: + """Stage type.""" + return self._element.type.value + + @property + def edges(self) -> List[List[UUID]]: + """Edge ID, source stage ID, target stage ID.""" + edges = [] + for edge in self._element.edges: + edges.append( + [edge.id, edge.source_stage_id, edge.target_stage_id] + ) + return edges \ No newline at end of file