minor fix of model files

Signed-off-by: Austin Liu <[email protected]>
austin362667 · Sep 7, 2024 · d074774 · d074774
1 parent b24cea5
commit d074774
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/flytekit/core/array_node.py b/flytekit/core/array_node.py
@@ -25,7 +25,7 @@ class ArrayNode:
     def __init__(
         self,
         target: LaunchPlan,
-        execution_mode: flyteidl.core.ArrayNode.ExecutionMode = flyteidl.core.ArrayNode.FULL_STATE,
+        execution_mode: flyteidl.core.ArrayNode.execution_mode = flyteidl.array_node.ExecutionMode.FullState,
         concurrency: Optional[int] = None,
         min_successes: Optional[int] = None,
         min_success_ratio: Optional[float] = None,
@@ -186,7 +186,7 @@ def concurrency(self) -> Optional[int]:
         return self._concurrency
 
     @property
-    def execution_mode(self) -> flyteidl.core.ArrayNode.ExecutionMode:
+    def execution_mode(self) -> flyteidl.array_node.ExecutionMode:
         return self._execution_mode
 
     def __call__(self, *args, **kwargs):

diff --git a/flytekit/models/execution.py b/flytekit/models/execution.py
@@ -186,7 +186,6 @@ def __init__(
         tags: Optional[typing.List[str]] = None,
         cluster_assignment: Optional[ClusterAssignment] = None,
         execution_cluster_label: Optional[matchable_resource.ExecutionClusterLabel] = None,
-        execution_cluster_label: Optional[ExecutionClusterLabel] = None,
     ):
         """
         :param flytekit.models.core.identifier.Identifier launch_plan: Launch plan unique identifier to execute

diff --git a/flytekit/models/types.py b/flytekit/models/types.py
@@ -280,6 +280,7 @@ def __init__(
         self._enum_type = enum_type
         self._union_type = union_type
         self._structured_dataset_type = structured_dataset_type
+        self._metadata = metadata
         self._structure = structure
         self._annotation = annotation
 
@@ -295,9 +296,9 @@ def __rich_repr__(self):
             sub = next(self.map_value_type.__rich_repr__())
             yield f"Dict[str, {sub}]"
         elif self.blob:
-            if self.blob.dimensionality == _types_pb2.BlobType.BlobDimensionality.SINGLE:
+            if self.blob.dimensionality == flyteidl.blob_type.BlobDimensionality.Single:
                 yield "File"
-            elif self.blob.dimensionality == _types_pb2.BlobType.BlobDimensionality.MULTIPART:
+            elif self.blob.dimensionality == flyteidl.blob_type.BlobDimensionality.Multipart:
                 yield "Directory"
             else:
                 yield "Unknown Blob Type"