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 643209c commit 02bf0a0Copy full SHA for 02bf0a0
pytensor/link/numba/dispatch/basic.py
@@ -39,6 +39,7 @@
39
from pytensor.tensor.slinalg import Solve
40
from pytensor.tensor.type import TensorType
41
from pytensor.tensor.type_other import MakeSlice, NoneConst
42
+from pytensor.typed_list import TypedListType
43
44
45
def global_numba_func(func):
@@ -121,6 +122,8 @@ def get_numba_type(
121
122
return CSCMatrixType(numba_dtype)
123
124
raise NotImplementedError()
125
+ elif isinstance(pytensor_type, TypedListType):
126
+ return numba.types.List(get_numba_type(pytensor_type.ttype))
127
else:
128
raise NotImplementedError(f"Numba type not implemented for {pytensor_type}")
129
0 commit comments