提交 d03acf4e authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Fix numba.njit stub used during Python-only coverage runs

上级 4bc063c0
...@@ -103,7 +103,7 @@ def eval_python_only(fn_inputs, fgraph, inputs): ...@@ -103,7 +103,7 @@ def eval_python_only(fn_inputs, fgraph, inputs):
return x return x
def njit_noop(*args, **kwargs): def njit_noop(*args, **kwargs):
if len(args) == 1: if len(args) == 1 and callable(args[0]):
return args[0] return args[0]
else: else:
return lambda x: x return lambda x: x
...@@ -118,9 +118,9 @@ def eval_python_only(fn_inputs, fgraph, inputs): ...@@ -118,9 +118,9 @@ def eval_python_only(fn_inputs, fgraph, inputs):
def inner_vec(*args): def inner_vec(*args):
if len(args) > nparams: if len(args) > nparams:
out = args[-1] out = args[-1]
out[:] = fn(*args[:nparams]) out[:] = np.vectorize(fn)(*args[:nparams])
else: else:
return fn(*args) return np.vectorize(fn)(*args)
return inner_vec return inner_vec
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论