Skip to content

Commit

Permalink
Plutopulp/pipeline failed state info (#446)
Browse files Browse the repository at this point in the history
* add failed state info to pipeline get schema

* bump version
  • Loading branch information
plutopulp authored Apr 9, 2024
1 parent 3372a21 commit 04c507e
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 16 deletions.
32 changes: 17 additions & 15 deletions pipeline/cloud/schemas/pipelines.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,21 +68,6 @@ class Pipeline(BaseModel):
extras: t.Optional[dict]


class PipelineGet(Pipeline):
id: str

created_at: datetime
updated_at: datetime

accelerators: t.Optional[t.List[Accelerator]]

cluster: PipelineClusterConfig | None = None

extras: t.Optional[dict]
#: The name of the scaling configuration
scaling_config: str | None = None


class PipelinePatch(BaseModel):
input_variables: t.Optional[t.List[IOVariable]]
output_variables: t.Optional[t.List[IOVariable]]
Expand Down Expand Up @@ -135,3 +120,20 @@ class PipelineScalingInfo(BaseModel):
current_replicas: int
desired_replicas: int
current_pipeline_states: dict[PipelineState, int]


class PipelineGet(Pipeline):
id: str

created_at: datetime
updated_at: datetime

accelerators: t.Optional[t.List[Accelerator]]

cluster: PipelineClusterConfig | None = None

extras: t.Optional[dict]
#: The name of the scaling configuration
scaling_config: str | None = None
#: Additional info attached when pipeline fails to load or startup
failed_state_info: t.Optional[PipelineContainerState]
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[tool.poetry]
name = "pipeline-ai"
version = "2.1.3"
version = "2.1.4"
description = "Pipelines for machine learning workloads."
authors = [
"Paul Hetherington <ph@mystic.ai>",
Expand Down

0 comments on commit 04c507e

Please sign in to comment.