From 228d4e8574f86f03af970c0be9785734944c04e2 Mon Sep 17 00:00:00 2001 From: italojohnny Date: Thu, 20 Jun 2024 21:56:53 -0300 Subject: [PATCH] fix union-attr error reported by mypy --- src/backend/base/langflow/graph/edge/base.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/graph/edge/base.py b/src/backend/base/langflow/graph/edge/base.py index c9fe43478..ca4856b23 100644 --- a/src/backend/base/langflow/graph/edge/base.py +++ b/src/backend/base/langflow/graph/edge/base.py @@ -68,11 +68,13 @@ class Edge: def _validate_handles(self, source, target) -> None: if self.target_handle.inputTypes is None: self.valid_handles = self.target_handle.type in self.source_handle.output_types - else: + + elif self.source_handle.output_types is not None: self.valid_handles = ( any(output_type in self.target_handle.inputTypes for output_type in self.source_handle.output_types) or self.target_handle.type in self.source_handle.output_types ) + if not self.valid_handles: logger.debug(self.source_handle) logger.debug(self.target_handle)