summaryrefslogtreecommitdiff
path: root/pylint/pyreverse/utils.py
diff options
context:
space:
mode:
authorDaniël van Noord <13665637+DanielNoord@users.noreply.github.com>2022-07-13 15:13:28 +0200
committerDaniël van Noord <13665637+DanielNoord@users.noreply.github.com>2022-07-13 15:24:05 +0200
commiteea753cb424027ea7583b0c62b80a53b26517701 (patch)
treea359a22447026945ad47c5a65746427423acd56a /pylint/pyreverse/utils.py
parent0502be5935aef736bf62f00dbdada3bd96afb05e (diff)
downloadpylint-git-eea753cb424027ea7583b0c62b80a53b26517701.tar.gz
Fix typing of most inference calls
Diffstat (limited to 'pylint/pyreverse/utils.py')
-rw-r--r--pylint/pyreverse/utils.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/pylint/pyreverse/utils.py b/pylint/pyreverse/utils.py
index b1be195e3..a1de38685 100644
--- a/pylint/pyreverse/utils.py
+++ b/pylint/pyreverse/utils.py
@@ -15,6 +15,7 @@ from typing import TYPE_CHECKING, Any, Callable, Optional, Tuple, Union
import astroid
from astroid import nodes
+from astroid.typing import InferenceResult
if TYPE_CHECKING:
from pylint.pyreverse.diagrams import ClassDiagram, PackageDiagram
@@ -210,7 +211,7 @@ def get_annotation(
return ann
-def infer_node(node: nodes.AssignAttr | nodes.AssignName) -> set[Any]:
+def infer_node(node: nodes.AssignAttr | nodes.AssignName) -> set[InferenceResult]:
"""Return a set containing the node annotation if it exists
otherwise return a set of the inferred types using the NodeNG.infer method.
"""