aboutsummaryrefslogtreecommitdiff
path: root/astroid/brain/brain_numpy_ma.py
diff options
context:
space:
mode:
Diffstat (limited to 'astroid/brain/brain_numpy_ma.py')
-rw-r--r--astroid/brain/brain_numpy_ma.py28
1 files changed, 28 insertions, 0 deletions
diff --git a/astroid/brain/brain_numpy_ma.py b/astroid/brain/brain_numpy_ma.py
new file mode 100644
index 00000000..8ae94659
--- /dev/null
+++ b/astroid/brain/brain_numpy_ma.py
@@ -0,0 +1,28 @@
+# Copyright (c) 2021 hippo91 <guillaume.peillex@gmail.com>
+
+# 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
+"""Astroid hooks for numpy ma module"""
+
+from astroid.brain.helpers import register_module_extender
+from astroid.builder import parse
+from astroid.manager import AstroidManager
+
+
+def numpy_ma_transform():
+ """
+ Infer the call of the masked_where function
+
+ :param node: node to infer
+ :param context: inference context
+ """
+ return parse(
+ """
+ import numpy.ma
+ def masked_where(condition, a, copy=True):
+ return numpy.ma.masked_array(a, mask=[])
+ """
+ )
+
+
+register_module_extender(AstroidManager(), "numpy.ma", numpy_ma_transform)