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 c62d733 commit dae7ff3Copy full SHA for dae7ff3
pytensor/link/numba/dispatch/basic.py
@@ -40,6 +40,7 @@
40
from pytensor.tensor.slinalg import Solve
41
from pytensor.tensor.type import TensorType
42
from pytensor.tensor.type_other import MakeSlice, NoneConst
43
+from pytensor.typed_list import TypedListType
44
45
46
def global_numba_func(func):
@@ -135,6 +136,8 @@ def get_numba_type(
135
136
return CSCMatrixType(numba_dtype)
137
138
raise NotImplementedError()
139
+ elif isinstance(pytensor_type, TypedListType):
140
+ return numba.types.List(get_numba_type(pytensor_type.ttype))
141
else:
142
raise NotImplementedError(f"Numba type not implemented for {pytensor_type}")
143
0 commit comments