pass turbo_registry flag on pipeline start upload (#473)

* pass turbo_registry flag on pipeline start upload * bump version:
mystic-ai · Jul 25, 2024 · bd5451d · bd5451d
1 parent b9794d1
commit bd5451d
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
@@ -34,6 +34,7 @@ class PipelineStartUpload(BaseModel):
     pipeline_name: str
     pipeline_tag: t.Optional[str]
     cluster: PipelineClusterConfig | None = None
+    turbo_registry: bool = False
 
 
 class PipelineStartUploadResponse(BaseModel):

diff --git a/pipeline/console/container/push.py b/pipeline/console/container/push.py
@@ -124,6 +124,7 @@ def push_container(namespace: Namespace):
                 pipeline_name=pipeline_name,
                 pipeline_tag=None,
                 cluster=pipeline_config.cluster,
+                turbo_registry=True if turbo_registry else False,
             ).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.4"
+version = "2.4.5"
 description = "Pipelines for machine learning workloads."
 authors = [
   "Paul Hetherington <ph@mystic.ai>",