We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7f61fb8 commit 7051623Copy full SHA for 7051623
tests/link/numba/test_basic.py
@@ -175,9 +175,18 @@ def inner_vec(*args):
175
else:
176
return wrap
177
178
+ def py_global_numba_func(func):
179
+ if hasattr(func, "py_func"):
180
+ return func.py_func
181
+ return func
182
+
183
mocks = [
184
mock.patch("numba.njit", njit_noop),
185
mock.patch("numba.vectorize", vectorize_noop),
186
+ mock.patch(
187
+ "pytensor.link.numba.dispatch.basic.global_numba_func",
188
+ py_global_numba_func,
189
+ ),
190
mock.patch(
191
"pytensor.link.numba.dispatch.basic.tuple_setitem", py_tuple_setitem
192
),
0 commit comments