summaryrefslogtreecommitdiff
path: root/numpy/lib/shape_base.pyi
diff options
context:
space:
mode:
authorBas van Beek <b.f.van.beek@vu.nl>2021-04-25 14:17:42 +0200
committerBas van Beek <b.f.van.beek@vu.nl>2021-04-26 16:41:04 +0200
commitbb34344242097b95db2be2e938c0ce14db225db3 (patch)
tree66924ba276a0d8e3613c75c1dcde29936a6fa79f /numpy/lib/shape_base.pyi
parent775d243be65f36564580ed5fcf038f384c61ce4d (diff)
downloadnumpy-bb34344242097b95db2be2e938c0ce14db225db3.tar.gz
ENH: Add improved placeholder annotations for `np.lib`
Diffstat (limited to 'numpy/lib/shape_base.pyi')
-rw-r--r--numpy/lib/shape_base.pyi24
1 files changed, 24 insertions, 0 deletions
diff --git a/numpy/lib/shape_base.pyi b/numpy/lib/shape_base.pyi
new file mode 100644
index 000000000..09edbcb6c
--- /dev/null
+++ b/numpy/lib/shape_base.pyi
@@ -0,0 +1,24 @@
+from typing import List
+
+from numpy.core.shape_base import vstack
+
+__all__: List[str]
+
+row_stack = vstack
+
+def take_along_axis(arr, indices, axis): ...
+def put_along_axis(arr, indices, values, axis): ...
+def apply_along_axis(func1d, axis, arr, *args, **kwargs): ...
+def apply_over_axes(func, a, axes): ...
+def expand_dims(a, axis): ...
+def column_stack(tup): ...
+def dstack(tup): ...
+def array_split(ary, indices_or_sections, axis=...): ...
+def split(ary, indices_or_sections, axis=...): ...
+def hsplit(ary, indices_or_sections): ...
+def vsplit(ary, indices_or_sections): ...
+def dsplit(ary, indices_or_sections): ...
+def get_array_prepare(*args): ...
+def get_array_wrap(*args): ...
+def kron(a, b): ...
+def tile(A, reps): ...