summaryrefslogtreecommitdiff
path: root/astroid
diff options
context:
space:
mode:
authorMarc Mueller <30130371+cdce8p@users.noreply.github.com>2021-04-24 12:44:06 +0200
committerPierre Sassoulas <pierre.sassoulas@gmail.com>2021-04-24 13:52:07 +0200
commitc064f1eba3ab3d9acee3873187488001b365d1f9 (patch)
tree74ccd4ba01494ade021c2d194f52cd9e9b581f91 /astroid
parent4d24273e9f5e710f3f43aa34f78d711e0cd87095 (diff)
downloadastroid-git-c064f1eba3ab3d9acee3873187488001b365d1f9.tar.gz
Add __init__ to astroid.brain package
Diffstat (limited to 'astroid')
-rw-r--r--astroid/__init__.py11
-rw-r--r--astroid/brain/__init__.py0
-rw-r--r--astroid/brain/brain_numpy_core_function_base.py4
-rw-r--r--astroid/brain/brain_numpy_core_multiarray.py4
-rw-r--r--astroid/brain/brain_numpy_core_numeric.py4
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(