Skip to content

Commit 7051623

Browse files
committed
Rollback changes to test_basic.py
1 parent 7f61fb8 commit 7051623

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

tests/link/numba/test_basic.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -175,9 +175,18 @@ def inner_vec(*args):
175175
else:
176176
return wrap
177177

178+
def py_global_numba_func(func):
179+
if hasattr(func, "py_func"):
180+
return func.py_func
181+
return func
182+
178183
mocks = [
179184
mock.patch("numba.njit", njit_noop),
180185
mock.patch("numba.vectorize", vectorize_noop),
186+
mock.patch(
187+
"pytensor.link.numba.dispatch.basic.global_numba_func",
188+
py_global_numba_func,
189+
),
181190
mock.patch(
182191
"pytensor.link.numba.dispatch.basic.tuple_setitem", py_tuple_setitem
183192
),

0 commit comments

Comments
 (0)