fix condition stop

This commit is contained in:
italojohnny 2024-06-24 18:37:31 -03:00 • committed by Gabriel Luiz Freitas Almeida
commit a8a8fbd351

View file

@ -96,7 +96,7 @@ class CustomComponent(BaseComponent):
def stop(self, output_name: str | None = None): def stop(self, output_name: str | None = None):
if not output_name and self.vertex and len(self.vertex.outputs) == 1: if not output_name and self.vertex and len(self.vertex.outputs) == 1:
output_name = self.vertex.outputs[0]["name"] output_name = self.vertex.outputs[0]["name"]
else: elif not output_name:
raise ValueError("You must specify an output name to call stop") raise ValueError("You must specify an output name to call stop")
if not self.vertex: if not self.vertex:
raise ValueError("Vertex is not set") raise ValueError("Vertex is not set")