summaryrefslogtreecommitdiff
path: root/astroid/brain
diff options
context:
space:
mode:
authorNick Drozd <nicholasdrozd@gmail.com>2022-12-15 16:00:12 -0600
committerGitHub <noreply@github.com>2022-12-15 23:00:12 +0100
commit39d9cafb8b39432957f4e640d5ef222b66a7096e (patch)
treecd1bbaec9764d9634918a361707130921a62301e /astroid/brain
parent6eb6bde7809c9ad2cc8d91c32804d9c8844207d6 (diff)
downloadastroid-git-39d9cafb8b39432957f4e640d5ef222b66a7096e.tar.gz
Misc type stuff (#1888)
Diffstat (limited to 'astroid/brain')
-rw-r--r--astroid/brain/brain_builtin_inference.py2
-rw-r--r--astroid/brain/brain_uuid.py2
-rw-r--r--astroid/brain/helpers.py11
3 files changed, 11 insertions, 4 deletions
diff --git a/astroid/brain/brain_builtin_inference.py b/astroid/brain/brain_builtin_inference.py
index a872b38c..61c0baa6 100644
--- a/astroid/brain/brain_builtin_inference.py
+++ b/astroid/brain/brain_builtin_inference.py
@@ -164,7 +164,7 @@ def _builtin_filter_predicate(node, builtin_name) -> bool:
return False
-def register_builtin_transform(transform, builtin_name):
+def register_builtin_transform(transform, builtin_name) -> None:
"""Register a new transform function for the given *builtin_name*.
The transform function must accept two parameters, a node and
diff --git a/astroid/brain/brain_uuid.py b/astroid/brain/brain_uuid.py
index f6ba8883..4890e101 100644
--- a/astroid/brain/brain_uuid.py
+++ b/astroid/brain/brain_uuid.py
@@ -8,7 +8,7 @@ from astroid.nodes.node_classes import Const
from astroid.nodes.scoped_nodes import ClassDef
-def _patch_uuid_class(node):
+def _patch_uuid_class(node: ClassDef) -> None:
# The .int member is patched using __dict__
node.locals["int"] = [Const(0, parent=node)]
diff --git a/astroid/brain/helpers.py b/astroid/brain/helpers.py
index d74f5950..56683c8e 100644
--- a/astroid/brain/helpers.py
+++ b/astroid/brain/helpers.py
@@ -2,11 +2,18 @@
# For details: https://github.com/PyCQA/astroid/blob/main/LICENSE
# Copyright (c) https://github.com/PyCQA/astroid/blob/main/CONTRIBUTORS.txt
+from __future__ import annotations
+
+from collections.abc import Callable
+
+from astroid.manager import AstroidManager
from astroid.nodes.scoped_nodes import Module
-def register_module_extender(manager, module_name, get_extension_mod):
- def transform(node):
+def register_module_extender(
+ manager: AstroidManager, module_name: str, get_extension_mod: Callable[[], Module]
+) -> None:
+ def transform(node: Module) -> None:
extension_module = get_extension_mod()
for name, objs in extension_module.locals.items():
node.locals[name] = objs