Skip to content

Commit d78e1a1

Browse files
aseyboldtricardoV94
authored andcommitted
Stop using deprecated generated_jit
1 parent 4077c47 commit d78e1a1

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

pytensor/link/numba/dispatch/basic.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -168,8 +168,12 @@ def in_seq_empty_tuple(x, y):
168168
enable_slice_boxing()
169169

170170

171-
@numba.generated_jit(nopython=True)
172171
def to_scalar(x):
172+
raise NotImplementedError()
173+
174+
175+
@numba.extending.overload(to_scalar)
176+
def impl_to_scalar(x):
173177
if isinstance(x, (numba.types.Number, numba.types.Boolean)):
174178
return lambda x: x
175179
elif isinstance(x, numba.types.Array):

0 commit comments

Comments
 (0)