diff options
| author | Marc Mueller <30130371+cdce8p@users.noreply.github.com> | 2021-04-24 12:44:06 +0200 |
|---|---|---|
| committer | Pierre Sassoulas <pierre.sassoulas@gmail.com> | 2021-04-24 13:52:07 +0200 |
| commit | c064f1eba3ab3d9acee3873187488001b365d1f9 (patch) | |
| tree | 74ccd4ba01494ade021c2d194f52cd9e9b581f91 /astroid | |
| parent | 4d24273e9f5e710f3f43aa34f78d711e0cd87095 (diff) | |
| download | astroid-git-c064f1eba3ab3d9acee3873187488001b365d1f9.tar.gz | |
Add __init__ to astroid.brain package
Diffstat (limited to 'astroid')
| -rw-r--r-- | astroid/__init__.py | 11 | ||||
| -rw-r--r-- | astroid/brain/__init__.py | 0 | ||||
| -rw-r--r-- | astroid/brain/brain_numpy_core_function_base.py | 4 | ||||
| -rw-r--r-- | astroid/brain/brain_numpy_core_multiarray.py | 4 | ||||
| -rw-r--r-- | astroid/brain/brain_numpy_core_numeric.py | 4 |
5 files changed, 10 insertions, 13 deletions
diff --git a/astroid/__init__.py b/astroid/__init__.py index 42a05539..0224dd7a 100644 --- a/astroid/__init__.py +++ b/astroid/__init__.py @@ -41,7 +41,8 @@ Main modules are: import enum import itertools import os -import sys +from importlib import import_module +from pathlib import Path import wrapt @@ -156,11 +157,7 @@ def register_module_extender(manager, module_name, get_extension_mod): # load brain plugins -BRAIN_MODULES_DIR = os.path.join(os.path.dirname(__file__), "brain") -if BRAIN_MODULES_DIR not in sys.path: - # add it to the end of the list so user path take precedence - sys.path.append(BRAIN_MODULES_DIR) -# load modules in this directory +BRAIN_MODULES_DIR = Path(__file__).with_name("brain") for module in os.listdir(BRAIN_MODULES_DIR): if module.endswith(".py"): - __import__(module[:-3]) + import_module(f"astroid.brain.{module[:-3]}") diff --git a/astroid/brain/__init__.py b/astroid/brain/__init__.py new file mode 100644 index 00000000..e69de29b --- /dev/null +++ b/astroid/brain/__init__.py diff --git a/astroid/brain/brain_numpy_core_function_base.py b/astroid/brain/brain_numpy_core_function_base.py index 1f875798..5901b068 100644 --- a/astroid/brain/brain_numpy_core_function_base.py +++ b/astroid/brain/brain_numpy_core_function_base.py @@ -10,10 +10,10 @@ import functools -from brain_numpy_utils import infer_numpy_member, looks_like_numpy_member - import astroid +from .brain_numpy_utils import infer_numpy_member, looks_like_numpy_member + METHODS_TO_BE_INFERRED = { "linspace": """def linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis=0): return numpy.ndarray([0, 0])""", diff --git a/astroid/brain/brain_numpy_core_multiarray.py b/astroid/brain/brain_numpy_core_multiarray.py index a33c3a27..2193c673 100644 --- a/astroid/brain/brain_numpy_core_multiarray.py +++ b/astroid/brain/brain_numpy_core_multiarray.py @@ -9,10 +9,10 @@ import functools -from brain_numpy_utils import infer_numpy_member, looks_like_numpy_member - import astroid +from .brain_numpy_utils import infer_numpy_member, looks_like_numpy_member + def numpy_core_multiarray_transform(): return astroid.parse( diff --git a/astroid/brain/brain_numpy_core_numeric.py b/astroid/brain/brain_numpy_core_numeric.py index 36c071d5..4df1e9a9 100644 --- a/astroid/brain/brain_numpy_core_numeric.py +++ b/astroid/brain/brain_numpy_core_numeric.py @@ -10,10 +10,10 @@ import functools -from brain_numpy_utils import infer_numpy_member, looks_like_numpy_member - import astroid +from .brain_numpy_utils import infer_numpy_member, looks_like_numpy_member + def numpy_core_numeric_transform(): return astroid.parse( |
