Skip to content

Commit adc66bb

Browse files
committed
numba careduce should return an array
1 parent 9828ec8 commit adc66bb

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

pytensor/link/numba/dispatch/elemwise.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -376,6 +376,7 @@ def careduce_maximum(input):
376376
careduce_def_src = f"""
377377
def {careduce_fn_name}({input_name}):
378378
{careduce_assign_lines}
379+
#return np.asarray({var_name})
379380
return {var_name}
380381
"""
381382

0 commit comments

Comments
 (0)