diff options
author | Jacob Walls <jacobtylerwalls@gmail.com> | 2023-05-14 16:33:33 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-05-14 16:33:33 -0400 |
commit | 835de848ac7cf51525d714f2f6ed07d789e09c54 (patch) | |
tree | 4f8cdb0f7d5b96708efd0c13e217e7e1e36261c8 /astroid/brain/brain_numpy_core_function_base.py | |
parent | 14eeb3fd64826c4a42a4c2f4edbac2476528dedc (diff) | |
download | astroid-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_function_base.py')
-rw-r--r-- | astroid/brain/brain_numpy_core_function_base.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/astroid/brain/brain_numpy_core_function_base.py b/astroid/brain/brain_numpy_core_function_base.py index bd218efa..f69826d5 100644 --- a/astroid/brain/brain_numpy_core_function_base.py +++ b/astroid/brain/brain_numpy_core_function_base.py @@ -6,7 +6,10 @@ 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, +) from astroid.inference_tip import inference_tip from astroid.manager import AstroidManager from astroid.nodes.node_classes import Attribute @@ -25,5 +28,5 @@ for func_name, func_src in METHODS_TO_BE_INFERRED.items(): AstroidManager().register_transform( Attribute, inference_tip(inference_function), - functools.partial(looks_like_numpy_member, func_name), + functools.partial(attribute_looks_like_numpy_member, func_name), ) |