fix return-value error reported by mypy

This commit is contained in:
italojohnny 2024-06-20 21:46:57 -03:00
commit 64e0968159

View file

@ -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