提交 d78e1a1a authored 作者: Adrian Seyboldt's avatar Adrian Seyboldt 提交者: Ricardo Vieira

Stop using deprecated generated_jit

上级 4077c47d
...@@ -168,8 +168,12 @@ def enable_slice_boxing(): ...@@ -168,8 +168,12 @@ def enable_slice_boxing():
enable_slice_boxing() enable_slice_boxing()
@numba.generated_jit(nopython=True)
def to_scalar(x): def to_scalar(x):
raise NotImplementedError()
@numba.extending.overload(to_scalar)
def impl_to_scalar(x):
if isinstance(x, (numba.types.Number, numba.types.Boolean)): if isinstance(x, (numba.types.Number, numba.types.Boolean)):
return lambda x: x return lambda x: x
elif isinstance(x, numba.types.Array): elif isinstance(x, numba.types.Array):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论