jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix union-attr error reported by mypy

Browse source
This commit is contained in:
italojohnny 2024-06-20 21:56:53 -03:00
commit 228d4e8574
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/graph/edge/base.py
View file

@ -68,11 +68,13 @@ class Edge:
def _validate_handles(self, source, target) -> None: def _validate_handles(self, source, target) -> None:
if self.target_handle.inputTypes is None: if self.target_handle.inputTypes is None:
self.valid_handles = self.target_handle.type in self.source_handle.output_types 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 = ( self.valid_handles = (
any(output_type in self.target_handle.inputTypes for output_type in self.source_handle.output_types) 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 or self.target_handle.type in self.source_handle.output_types
) )
if not self.valid_handles: if not self.valid_handles:
logger.debug(self.source_handle) logger.debug(self.source_handle)
logger.debug(self.target_handle) logger.debug(self.target_handle)