summaryrefslogtreecommitdiff
path: root/numpy/typing
diff options
context:
space:
mode:
authorBas van Beek <43369155+BvB93@users.noreply.github.com>2022-07-14 13:16:59 +0200
committerBas van Beek <43369155+BvB93@users.noreply.github.com>2022-07-14 13:16:59 +0200
commit79a895d6be4da326a1d7c5a018ed3ba62adaf8e7 (patch)
tree05129c02706980c670e7e9a9f382df1f354e818c /numpy/typing
parent37846563156b3ed8287fc94d6a7d107a316c5ecf (diff)
downloadnumpy-79a895d6be4da326a1d7c5a018ed3ba62adaf8e7.tar.gz
TYP,MAINT: Allow `einsum` subscripts to be passed via integer array-likes
Diffstat (limited to 'numpy/typing')
-rw-r--r--numpy/typing/tests/data/reveal/einsumfunc.pyi3
1 files changed, 3 insertions, 0 deletions
diff --git a/numpy/typing/tests/data/reveal/einsumfunc.pyi b/numpy/typing/tests/data/reveal/einsumfunc.pyi
index 3c7146ada..d5f930149 100644
--- a/numpy/typing/tests/data/reveal/einsumfunc.pyi
+++ b/numpy/typing/tests/data/reveal/einsumfunc.pyi
@@ -30,3 +30,6 @@ reveal_type(np.einsum_path("i,i->i", AR_LIKE_f, AR_LIKE_f)) # E: Tuple[builtins
reveal_type(np.einsum_path("i,i->i", AR_LIKE_c, AR_LIKE_c)) # E: Tuple[builtins.list[Any], builtins.str]
reveal_type(np.einsum_path("i,i->i", AR_LIKE_b, AR_LIKE_i)) # E: Tuple[builtins.list[Any], builtins.str]
reveal_type(np.einsum_path("i,i,i,i->i", AR_LIKE_b, AR_LIKE_u, AR_LIKE_i, AR_LIKE_c)) # E: Tuple[builtins.list[Any], builtins.str]
+
+reveal_type(np.einsum([[1, 1], [1, 1]], AR_LIKE_i, AR_LIKE_i)) # E: Any
+reveal_type(np.einsum_path([[1, 1], [1, 1]], AR_LIKE_i, AR_LIKE_i)) # E: Tuple[builtins.list[Any], builtins.str]