truefoundry 0.5.6rc1__tar.gz → 0.5.6rc2__tar.gz

{truefoundry-0.5.6rc1 → truefoundry-0.5.6rc2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: truefoundry
-Version: 0.5.6rc1
+Version: 0.5.6rc2
 Summary: TrueFoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com

{truefoundry-0.5.6rc1 → truefoundry-0.5.6rc2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truefoundry"
-version = "0.5.6rc1"
+version = "0.5.6rc2"
 description = "TrueFoundry CLI"
 authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
 readme = "README.md"

{truefoundry-0.5.6rc1 → truefoundry-0.5.6rc2}/truefoundry/deploy/v2/lib/deploy_workflow.py

@@ -1,7 +1,7 @@
 import os
 import sys
 from pathlib import Path
-from typing import Dict, List, Optional, Union
+from typing import Callable, Dict, List, Optional, Union
 
 import requirements
 from flytekit.configuration import (
@@ -37,13 +37,16 @@ from truefoundry.workflow.workflow import (
 
 
 def _handle_code_upload_for_workflow(
-    workflow: auto_gen_models.Workflow, workspace_fqn: str
+    workflow: auto_gen_models.Workflow,
+    workspace_fqn: str,
+    upload_code_package: Callable[[str, str, str], str],
 ) -> auto_gen_models.Workflow:
     new_workflow = workflow.copy(deep=True)
     new_workflow.source = local_source_to_remote_source(
         local_source=workflow.source,
         workspace_fqn=workspace_fqn,
         component_name=workflow.name,
+        upload_code_package=upload_code_package,
     )
     return new_workflow
 
@@ -283,12 +286,14 @@ def deploy_workflow(
     logger.info(
         f"Deploying workflow {workflow.name} to workspace {workspace_fqn} ({workspace_id})"
     )
+    client = ServiceFoundryServiceClient()
 
     workflow = _handle_code_upload_for_workflow(
-        workflow=workflow, workspace_fqn=workspace_fqn
+        workflow=workflow,
+        workspace_fqn=workspace_fqn,
+        upload_code_package=client.upload_code_package,
     )
 
-    client = ServiceFoundryServiceClient()
     deployment = client.deploy_application(
         workspace_id=workspace_id, application=workflow
     )