diff --git a/flytekit/remote/entities.py b/flytekit/remote/entities.py index f4f08921f9..00e9ba40de 100644 --- a/flytekit/remote/entities.py +++ b/flytekit/remote/entities.py @@ -348,25 +348,19 @@ def promote_from_model(cls, model: _workflow_model.GateNode): class FlyteArrayNode(_workflow_model.ArrayNode): - def __init__(self, node, parallelism, min_successes, min_success_ratio, flyte_task_node=None, flyte_workflow_node=None): + def __init__(self, node, parallelism, min_successes, min_success_ratio, flyte_task_node=None): super().__init__(node, parallelism, min_successes, min_success_ratio) self._flyte_task_node = flyte_task_node - self._flyte_workflow_node = flyte_workflow_node @property def flyte_task_node(self): return self._flyte_task_node - @property - def flyte_workflow_node(self): - return self._flyte_workflow_node - @classmethod def promote_from_model( cls, model: _workflow_model.ArrayNode, flyte_task_node: Optional[FlyteTaskNode] = None, - flyte_workflow_node: Optional[FlyteWorkflowNode] = None, ): return cls( node=model._node, @@ -374,7 +368,6 @@ def promote_from_model( min_successes=model._min_successes, min_success_ratio=model._min_success_ratio, flyte_task_node=flyte_task_node, - flyte_workflow_node=flyte_workflow_node, ) @@ -504,18 +497,6 @@ def promote_from_model( f"Remote Workflow closure does not have task with id {model.array_node.node.task_node.reference_id}." ) flyte_array_node = FlyteArrayNode.promote_from_model(model.array_node, flyte_task_node=cls._promote_task_node(tasks[model.array_node.node.task_node.reference_id])) - # map over launch plan - elif model.array_node.node.workflow_node is not None: - workflow_node, converted_sub_workflows = cls._promote_workflow_node( - model.array_node.node.workflow_node, - sub_workflows, - node_launch_plans, - tasks, - converted_sub_workflows, - ) - flyte_array_node = FlyteArrayNode.promote_from_model( - model.array_node, flyte_workflow_node=workflow_node - ) else: raise _system_exceptions.FlyteSystemException( "Array node must have either task or workflow node specified"