Skip to content

Commit a5de97c

Browse files
Fix AllocDiag and Tri gufunc signatures
1 parent 0a0da09 commit a5de97c

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

aesara/tensor/basic.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -980,9 +980,9 @@ class Tri(Op):
980980

981981
gufunc_sig = (
982982
(
983-
("n",),
984-
("m",),
985-
("k",),
983+
(),
984+
(),
985+
(),
986986
),
987987
(("n", "m"),),
988988
)
@@ -3508,7 +3508,7 @@ class AllocDiag(Op):
35083508
It does the inverse of `ExtractDiag`.
35093509
"""
35103510

3511-
gufunc_sig = ((("m",),), (("m", "m"),))
3511+
gufunc_sig = (((),), (("m", "m"),))
35123512
__props__ = ("offset", "axis1", "axis2", "gufunc_sig")
35133513

35143514
def __init__(self, offset=0, axis1=0, axis2=1):

0 commit comments

Comments
 (0)