update flyteidl

Signed-off-by: Iaroslav Ciupin <iaroslav@union.ai>
flyteorg · Sep 30, 2022 · 8e6b944 · 8e6b944
1 parent e191524
commit 8e6b944
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/dev-requirements.txt b/dev-requirements.txt
@@ -121,7 +121,7 @@ docstring-parser==0.14.1
     #   flytekit
 filelock==3.8.0
     # via virtualenv
-flyteidl==git+https://github.com/flyteorg/flyteidl.git@cluster-pools-assignment
+flyteidl==1.1.19
     # via
     #   -c requirements.txt
     #   flytekit

diff --git a/flytekit/models/execution.py b/flytekit/models/execution.py
@@ -4,7 +4,7 @@
 import typing
 
 import flyteidl
-import flyteidl.admin as _admin_pb2
+import flyteidl.admin.cluster_assignment_pb2 as _cluster_assignment_pb2
 import flyteidl.admin.execution_pb2 as _execution_pb2
 import flyteidl.admin.node_execution_pb2 as _node_execution_pb2
 import flyteidl.admin.task_execution_pb2 as _task_execution_pb2
@@ -335,7 +335,7 @@ def to_flyte_idl(self):
         """
         :rtype: flyteidl.admin.ClusterAssignment
         """
-        return _admin_pb2.ClusterAssignment(
+        return _cluster_assignment_pb2.ClusterAssignment(
             cluster_pool=self.cluster_pool,
         )
 

diff --git a/requirements.txt b/requirements.txt
@@ -48,7 +48,7 @@ docker-image-py==0.1.12
     # via flytekit
 docstring-parser==0.14.1
     # via flytekit
-flyteidl==git+https://github.com/flyteorg/flyteidl.git@cluster-pools-assignment
+flyteidl==1.1.19
     # via flytekit
 googleapis-common-protos==1.56.4
     # via