Removing declaration of zeros_like (pass). Filtering tuple and list from types inferred by call to full_like and ones_like

1 files changed, 12 insertions, 1 deletions

diff --git a/astroid/brain/brain_numpy.py b/astroid/brain/brain_numpy.py
index 1d0e42f9..919c9607 100644
--- a/astroid/brain/brain_numpy.py
+++ b/astroid/brain/brain_numpy.py
@@ -44,7 +44,6 @@ def numpy_random_mtrand_transform():
     def power(a, size=None): return any
     def rand(*args): return any
     def randint(low, high=None, size=None, dtype='l'): pass
-    def zeros_like(a, dtype=None, order='K', subok=True): pass
     def randn(*args): return any
     def random_integers(low, high=None, size=None): return any
     def random_sample(size=None): return any
@@ -552,6 +551,18 @@ astroid.MANAGER.register_transform(
     functools.partial(_looks_like_numpy_function, "zeros_like", "numpy.core.numeric"),
 )
 
+astroid.MANAGER.register_transform(
+    astroid.FunctionDef,
+    _replace_numpy_function_infer_call_result,
+    functools.partial(_looks_like_numpy_function, "full_like", "numpy.core.numeric"),
+)
+
+astroid.MANAGER.register_transform(
+    astroid.FunctionDef,
+    _replace_numpy_function_infer_call_result,
+    functools.partial(_looks_like_numpy_function, "ones_like", "numpy.core.numeric"),
+)
+
 astroid.register_module_extender(
     astroid.MANAGER, "numpy.core.umath", numpy_core_umath_transform
 )