Skip to content

Commit 173a5ba

Browse files
committed
Bring back
1 parent 7051623 commit 173a5ba

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

pytensor/link/numba/dispatch/basic.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,16 @@
4343
from pytensor.tensor.type_other import MakeSlice, NoneConst
4444

4545

46+
def global_numba_func(func):
47+
"""Use to return global numba functions in numba_funcify_*.
48+
49+
50+
51+
This allows tests to remove the compilation using mock.
52+
"""
53+
return func
54+
55+
4656
def numba_njit(*args, fastmath=None, **kwargs):
4757
kwargs.setdefault("cache", config.numba__cache)
4858
kwargs.setdefault("no_cpython_wrapper", True)

0 commit comments

Comments
 (0)