diff options
Diffstat (limited to 'astroid/brain/brain_numpy_ma.py')
-rw-r--r-- | astroid/brain/brain_numpy_ma.py | 28 |
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) |