summaryrefslogtreecommitdiff
path: root/astroid/brain/brain_numpy_core_multiarray.py
diff options
context:
space:
mode:
authorJacob Walls <jacobtylerwalls@gmail.com>2023-05-14 16:33:33 -0400
committerGitHub <noreply@github.com>2023-05-14 16:33:33 -0400
commit835de848ac7cf51525d714f2f6ed07d789e09c54 (patch)
tree4f8cdb0f7d5b96708efd0c13e217e7e1e36261c8 /astroid/brain/brain_numpy_core_multiarray.py
parent14eeb3fd64826c4a42a4c2f4edbac2476528dedc (diff)
downloadastroid-git-835de848ac7cf51525d714f2f6ed07d789e09c54.tar.gz
Improve performance of `looks_like_numpy_member()` (#2178)
Avoids 32% of the calls to isinstance() when linting astroid
Diffstat (limited to 'astroid/brain/brain_numpy_core_multiarray.py')
-rw-r--r--astroid/brain/brain_numpy_core_multiarray.py10
1 files changed, 7 insertions, 3 deletions
diff --git a/astroid/brain/brain_numpy_core_multiarray.py b/astroid/brain/brain_numpy_core_multiarray.py
index 4b2fe63c..e9c7bacf 100644
--- a/astroid/brain/brain_numpy_core_multiarray.py
+++ b/astroid/brain/brain_numpy_core_multiarray.py
@@ -6,7 +6,11 @@
import functools
-from astroid.brain.brain_numpy_utils import infer_numpy_member, looks_like_numpy_member
+from astroid.brain.brain_numpy_utils import (
+ attribute_looks_like_numpy_member,
+ infer_numpy_member,
+ name_looks_like_numpy_member,
+)
from astroid.brain.helpers import register_module_extender
from astroid.builder import parse
from astroid.inference_tip import inference_tip
@@ -91,10 +95,10 @@ for method_name, function_src in METHODS_TO_BE_INFERRED.items():
AstroidManager().register_transform(
Attribute,
inference_tip(inference_function),
- functools.partial(looks_like_numpy_member, method_name),
+ functools.partial(attribute_looks_like_numpy_member, method_name),
)
AstroidManager().register_transform(
Name,
inference_tip(inference_function),
- functools.partial(looks_like_numpy_member, method_name),
+ functools.partial(name_looks_like_numpy_member, method_name),
)