Skip to content

Commit de75bba

Browse files
committed
More readable names for numba set_subtensor funcs
1 parent d878e6a commit de75bba

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pytensor/link/numba/dispatch/subtensor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,10 +77,10 @@ def convert_indices(indices, entry):
7777
y_name = input_names[1]
7878

7979
if op.set_instead_of_inc:
80-
function_name = "setsubtensor"
80+
function_name = "set_subtensor"
8181
index_body = f"z[indices] = {y_name}"
8282
else:
83-
function_name = "incsubtensor"
83+
function_name = "inc_subtensor"
8484
index_body = f"z[indices] += {y_name}"
8585
else:
8686
function_name = "subtensor"

0 commit comments

Comments
 (0)