provide accelerators pipeline start upload (#474)

* add accelerators to PipelineStartUpload * bump version
mystic-ai · Jul 26, 2024 · 3aa9aab · 3aa9aab
1 parent bd5451d
commit 3aa9aab
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/pipeline/cloud/schemas/pipelines.py b/pipeline/cloud/schemas/pipelines.py
@@ -35,6 +35,7 @@ class PipelineStartUpload(BaseModel):
     pipeline_tag: t.Optional[str]
     cluster: PipelineClusterConfig | None = None
     turbo_registry: bool = False
+    accelerators: list[Accelerator] = []
 
 
 class PipelineStartUploadResponse(BaseModel):

diff --git a/pipeline/console/container/push.py b/pipeline/console/container/push.py
@@ -125,6 +125,7 @@ def push_container(namespace: Namespace):
                 pipeline_tag=None,
                 cluster=pipeline_config.cluster,
                 turbo_registry=True if turbo_registry else False,
+                accelerators=pipeline_config.accelerators,
             ).dict(),
         )
         start_upload_dict = start_upload_response.json()

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pipeline-ai"
-version = "2.4.5"
+version = "2.4.6"
 description = "Pipelines for machine learning workloads."
 authors = [
   "Paul Hetherington <ph@mystic.ai>",