Skip to content

Commit 0e9f37c

Browse files
committed
Remove unused numba_vectorize
1 parent 934306f commit 0e9f37c

File tree

2 files changed

+0
-11
lines changed

2 files changed

+0
-11
lines changed

pytensor/link/numba/dispatch/basic.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -87,13 +87,6 @@ def numba_njit(*args, fastmath=None, **kwargs):
8787
return numba.njit(*args, fastmath=fastmath, **kwargs)
8888

8989

90-
def numba_vectorize(*args, **kwargs):
91-
if len(args) > 0 and callable(args[0]):
92-
return numba.vectorize(*args[1:], cache=config.numba__cache, **kwargs)(args[0])
93-
94-
return numba.vectorize(*args, cache=config.numba__cache, **kwargs)
95-
96-
9790
def get_numba_type(
9891
pytensor_type: Type,
9992
layout: str = "A",

tests/link/numba/test_basic.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,6 @@ def py_global_numba_func(func):
182182

183183
mocks = [
184184
mock.patch("numba.njit", njit_noop),
185-
mock.patch("numba.vectorize", vectorize_noop),
186185
mock.patch(
187186
"pytensor.link.numba.dispatch.basic.global_numba_func",
188187
py_global_numba_func,
@@ -191,9 +190,6 @@ def py_global_numba_func(func):
191190
"pytensor.link.numba.dispatch.basic.tuple_setitem", py_tuple_setitem
192191
),
193192
mock.patch("pytensor.link.numba.dispatch.basic.numba_njit", njit_noop),
194-
mock.patch(
195-
"pytensor.link.numba.dispatch.basic.numba_vectorize", vectorize_noop
196-
),
197193
mock.patch(
198194
"pytensor.link.numba.dispatch.basic.direct_cast", lambda x, dtype: x
199195
),

0 commit comments

Comments
 (0)