From 64e0968159f2d42bf7264a7dc703008ec8859601 Mon Sep 17 00:00:00 2001 From: italojohnny Date: Thu, 20 Jun 2024 21:46:57 -0300 Subject: [PATCH] fix return-value error reported by mypy --- src/backend/base/langflow/graph/vertex/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/graph/vertex/base.py b/src/backend/base/langflow/graph/vertex/base.py index 86a65b9cf..4828aa92b 100644 --- a/src/backend/base/langflow/graph/vertex/base.py +++ b/src/backend/base/langflow/graph/vertex/base.py @@ -5,7 +5,7 @@ import os import traceback import types from enum import Enum -from typing import TYPE_CHECKING, Any, AsyncIterator, Callable, Dict, Iterator, List, Mapping, Optional +from typing import TYPE_CHECKING, Set, Any, AsyncIterator, Callable, Dict, Iterator, List, Mapping, Optional from loguru import logger @@ -152,7 +152,7 @@ class Vertex: return [edge for edge in self.edges if edge.target_id == self.id] @property - def edges_source_names(self) -> List[str]: + def edges_source_names(self) -> Set[str | None]: return {edge.source_handle.name for edge in self.edges} @property