Fix is_task attribute in CustomComponentVertex

constructor
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-28 22:02:55 -03:00
commit c14f95c100

View file

@ -290,7 +290,7 @@ class OutputParserVertex(Vertex):
class CustomComponentVertex(Vertex):
def __init__(self, data: Dict, graph):
super().__init__(data, graph=graph, base_type="custom_components", is_task=True)
super().__init__(data, graph=graph, base_type="custom_components", is_task=False)
def _built_object_repr(self):
if self.task_id and self.is_task: