File tree Expand file tree Collapse file tree 2 files changed +0
-17
lines changed
pytensor/link/numba/dispatch Expand file tree Collapse file tree 2 files changed +0
-17
lines changed Original file line number Diff line number Diff line change 4242from pytensor .tensor .type_other import MakeSlice , NoneConst
4343
4444
45- def global_numba_func (func ):
46- """Use to return global numba functions in numba_funcify_*.
47-
48- This allows tests to remove the compilation using mock.
49- """
50- return func
51-
52-
5345def numba_njit (* args , fastmath = None , ** kwargs ):
5446 kwargs .setdefault ("cache" , config .numba__cache )
5547 kwargs .setdefault ("no_cpython_wrapper" , True )
Original file line number Diff line number Diff line change @@ -175,17 +175,8 @@ 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-
183178 mocks = [
184179 mock .patch ("numba.njit" , njit_noop ),
185- mock .patch (
186- "pytensor.link.numba.dispatch.basic.global_numba_func" ,
187- py_global_numba_func ,
188- ),
189180 mock .patch (
190181 "pytensor.link.numba.dispatch.basic.tuple_setitem" , py_tuple_setitem
191182 ),
You can’t perform that action at this time.
0 commit comments