MOONy

✎ EDIT FILE

EDITING: brain_numpy_core_multiarray.py

# Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
# For details: https://github.com/PyCQA/astroid/blob/main/LICENSE
# Copyright (c) https://github.com/PyCQA/astroid/blob/main/CONTRIBUTORS.txt

"""Astroid hooks for numpy.core.multiarray module."""

import functools

from astroid.brain.brain_numpy_utils import infer_numpy_member, looks_like_numpy_member
from astroid.brain.helpers import register_module_extender
from astroid.builder import parse
from astroid.inference_tip import inference_tip
from astroid.manager import AstroidManager
from astroid.nodes.node_classes import Attribute, Name

def numpy_core_multiarray_transform():
    return parse(
        """
    # different functions defined in multiarray.py
    def inner(a, b):
        return numpy.ndarray([0, 0])

def vdot(a, b):
        return numpy.ndarray([0, 0])
        """
    )

register_module_extender(
    AstroidManager(), "numpy.core.multiarray", numpy_core_multiarray_transform
)

METHODS_TO_BE_INFERRED = {
    "array": """def array(object, dtype=None, copy=True, order='K', subok=False, ndmin=0):
            return numpy.ndarray([0, 0])""",
    "dot": """def dot(a, b, out=None):
            return numpy.ndarray([0, 0])""",
    "empty_like": """def empty_like(a, dtype=None, order='K', subok=True):
            return numpy.ndarray((0, 0))""",
    "concatenate": """def concatenate(arrays, axis=None, out=None):
            return numpy.ndarray((0, 0))""",
    "where": """def where(condition, x=None, y=None):
            return numpy.ndarray([0, 0])""",
    "empty": """def empty(shape, dtype=float, order='C'):
            return numpy.ndarray([0, 0])""",
    "bincount": """def bincount(x, weights=None, minlength=0):
            return numpy.ndarray([0, 0])""",
    "busday_count": """def busday_count(
        begindates, enddates, weekmask='1111100', holidays=[], busdaycal=None, out=None
    ):
        return numpy.ndarray([0, 0])""",
    "busday_offset": """def busday_offset(
        dates, offsets, roll='raise', weekmask='1111100', holidays=None,
        busdaycal=None, out=None
    ):
        return numpy.ndarray([0, 0])""",
    "can_cast": """def can_cast(from_, to, casting='safe'):
            return True""",
    "copyto": """def copyto(dst, src, casting='same_kind', where=True):
            return None""",
    "datetime_as_string": """def datetime_as_string(arr, unit=None, timezone='naive', casting='same_kind'):
            return numpy.ndarray([0, 0])""",
    "is_busday": """def is_busday(dates, weekmask='1111100', holidays=None, busdaycal=None, out=None):
            return numpy.ndarray([0, 0])""",
    "lexsort": """def lexsort(keys, axis=-1):
            return numpy.ndarray([0, 0])""",
    "may_share_memory": """def may_share_memory(a, b, max_work=None):
            return True""",
    # Not yet available because dtype is not yet present in those brains
    #     "min_scalar_type": """def min_scalar_type(a):
    #             return numpy.dtype('int16')""",
    "packbits": """def packbits(a, axis=None, bitorder='big'):
            return numpy.ndarray([0, 0])""",
    # Not yet available because dtype is not yet present in those brains
    #     "result_type": """def result_type(*arrays_and_dtypes):
    #             return numpy.dtype('int16')""",
    "shares_memory": """def shares_memory(a, b, max_work=None):
            return True""",
    "unpackbits": """def unpackbits(a, axis=None, count=None, bitorder='big'):
            return numpy.ndarray([0, 0])""",
    "unravel_index": """def unravel_index(indices, shape, order='C'):
            return (numpy.ndarray([0, 0]),)""",
    "zeros": """def zeros(shape, dtype=float, order='C'):
            return numpy.ndarray([0, 0])""",
}

for method_name, function_src in METHODS_TO_BE_INFERRED.items():
    inference_function = functools.partial(infer_numpy_member, function_src)
    AstroidManager().register_transform(
        Attribute,
        inference_tip(inference_function),
        functools.partial(looks_like_numpy_member, method_name),
    )
    AstroidManager().register_transform(
        Name,
        inference_tip(inference_function),
        functools.partial(looks_like_numpy_member, method_name),
    )

CANCEL

Name	Type	Size	Modified	Actions
↩ ..	DIR	—	—
📁 __pycache__/	DIR	—	2026-09-08 23:51
📄 __init__.py	PY	0 B	2026-08-21 11:54	EDIT
📄 brain_argparse.py	PY	1.5 KB	2026-08-21 11:54	EDIT
📄 brain_attrs.py	PY	3 KB	2026-08-21 11:54	EDIT
📄 brain_boto3.py	PY	1012 B	2026-08-21 11:54	EDIT
📄 brain_builtin_inference.py	PY	33.5 KB	2026-08-21 11:54	EDIT
📄 brain_collections.py	PY	4.3 KB	2026-08-21 11:54	EDIT
📄 brain_crypt.py	PY	863 B	2026-08-21 11:54	EDIT
📄 brain_ctypes.py	PY	2.6 KB	2026-08-21 11:54	EDIT
📄 brain_curses.py	PY	3.4 KB	2026-08-21 11:54	EDIT
📄 brain_dataclasses.py	PY	21.6 KB	2026-08-21 11:54	EDIT
📄 brain_dateutil.py	PY	767 B	2026-08-21 11:54	EDIT
📄 brain_fstrings.py	PY	2.4 KB	2026-08-21 11:54	EDIT
📄 brain_functools.py	PY	5.8 KB	2026-08-21 11:54	EDIT
📄 brain_gi.py	PY	7.4 KB	2026-08-21 11:54	EDIT
📄 brain_hashlib.py	PY	2.8 KB	2026-08-21 11:54	EDIT
📄 brain_http.py	PY	10.4 KB	2026-08-21 11:54	EDIT
📄 brain_hypothesis.py	PY	1.7 KB	2026-08-21 11:54	EDIT
📄 brain_io.py	PY	1.5 KB	2026-08-21 11:54	EDIT
📄 brain_mechanize.py	PY	2.6 KB	2026-08-21 11:54	EDIT
📄 brain_multiprocessing.py	PY	3.1 KB	2026-08-21 11:54	EDIT
📄 brain_namedtuple_enum.py	PY	22.2 KB	2026-08-21 11:54	EDIT
📄 brain_nose.py	PY	2.3 KB	2026-08-21 11:54	EDIT
📄 brain_numpy_core_einsumfunc.py	PY	825 B	2026-08-21 11:54	EDIT
📄 brain_numpy_core_fromnumeric.py	PY	732 B	2026-08-21 11:54	EDIT
📄 brain_numpy_core_function_base.py	PY	1.3 KB	2026-08-21 11:54	EDIT
📄 brain_numpy_core_multiarray.py	PY	4.1 KB	2026-08-21 11:54	EDIT
📄 brain_numpy_core_numeric.py	PY	1.6 KB	2026-08-21 11:54	EDIT
📄 brain_numpy_core_numerictypes.py	PY	8.3 KB	2026-08-21 11:54	EDIT
📄 brain_numpy_core_umath.py	PY	4.8 KB	2026-08-21 11:54	EDIT
📄 brain_numpy_ma.py	PY	896 B	2026-08-21 11:54	EDIT
📄 brain_numpy_ndarray.py	PY	8.8 KB	2026-08-21 11:54	EDIT
📄 brain_numpy_random_mtrand.py	PY	3.4 KB	2026-08-21 11:54	EDIT
📄 brain_numpy_utils.py	PY	2.6 KB	2026-08-21 11:54	EDIT
📄 brain_pathlib.py	PY	1.5 KB	2026-08-21 11:54	EDIT
📄 brain_pkg_resources.py	PY	2.1 KB	2026-08-21 11:54	EDIT
📄 brain_pytest.py	PY	2.2 KB	2026-08-21 11:54	EDIT
📄 brain_qt.py	PY	2.7 KB	2026-08-21 11:54	EDIT
📄 brain_random.py	PY	2.8 KB	2026-08-21 11:54	EDIT
📄 brain_re.py	PY	2.8 KB	2026-08-21 11:54	EDIT
📄 brain_regex.py	PY	3.3 KB	2026-08-21 11:54	EDIT
📄 brain_responses.py	PY	1.8 KB	2026-08-21 11:54	EDIT
📄 brain_scipy_signal.py	PY	2.2 KB	2026-08-21 11:54	EDIT
📄 brain_signal.py	PY	3.8 KB	2026-08-21 11:54	EDIT
📄 brain_six.py	PY	7.4 KB	2026-08-21 11:54	EDIT
📄 brain_sqlalchemy.py	PY	1009 B	2026-08-21 11:54	EDIT
📄 brain_ssl.py	PY	6.4 KB	2026-08-21 11:54	EDIT
📄 brain_subprocess.py	PY	2.9 KB	2026-08-21 11:54	EDIT
📄 brain_threading.py	PY	870 B	2026-08-21 11:54	EDIT
📄 brain_type.py	PY	2.4 KB	2026-08-21 11:54	EDIT
📄 brain_typing.py	PY	14 KB	2026-08-21 11:54	EDIT
📄 brain_unittest.py	PY	1.1 KB	2026-08-21 11:54	EDIT
📄 brain_uuid.py	PY	667 B	2026-08-21 11:54	EDIT
📄 helpers.py	PY	908 B	2026-08-21 11:54	EDIT