adjust from_flyte_idl

Signed-off-by: Iaroslav Ciupin <[email protected]>
flyteorg · Sep 21, 2023 · 4f97c87 · 4f97c87
1 parent 80720d0
commit 4f97c87
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/flytekit/models/execution.py b/flytekit/models/execution.py
@@ -342,6 +342,7 @@ def from_flyte_idl(cls, p):
             overwrite_cache=p.overwrite_cache,
             envs=_common_models.Envs.from_flyte_idl(p.envs) if p.HasField("envs") else None,
             tags=p.tags,
+            cluster_assignment=ClusterAssignment(cluster_pool=p.cluster_pool_name) if p.cluster_pool_name else None,
         )