| 12345678910111213141516171819202122232425262728293031 |
- # 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 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 various numpy.ma functions.
- :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=[])
- def masked_invalid(a, copy=True):
- return numpy.ma.masked_array(a, mask=[])
- """
- )
- register_module_extender(AstroidManager(), "numpy.ma", numpy_ma_transform)
|
