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 1e79035 commit 6f6e8bbCopy full SHA for 6f6e8bb
pytensor/link/numba/dispatch/basic.py
@@ -48,6 +48,7 @@
48
)
49
from pytensor.tensor.type import TensorType
50
from pytensor.tensor.type_other import MakeSlice, NoneConst
51
+from pytensor.typed_list import TypedListType
52
53
54
def global_numba_func(func):
@@ -130,6 +131,8 @@ def get_numba_type(
130
131
return CSCMatrixType(numba_dtype)
132
133
raise NotImplementedError()
134
+ elif isinstance(pytensor_type, TypedListType):
135
+ return numba.types.List(get_numba_type(pytensor_type.ttype))
136
else:
137
raise NotImplementedError(f"Numba type not implemented for {pytensor_type}")
138
0 commit comments