mirror of https://github.com/Nonannet/copapy.git
type checker annotation fixed for jit decorator function
This commit is contained in:
parent
ce15e83c2b
commit
a07332f3a4
|
|
@ -32,7 +32,7 @@ def jit(func: Callable[..., TRet]) -> Callable[..., TRet]:
|
||||||
tg = Target()
|
tg = Target()
|
||||||
inputs = tuple(
|
inputs = tuple(
|
||||||
tuple(value(ai) for ai in a) if isinstance(a, Iterable) else value(a) for a in args)
|
tuple(value(ai) for ai in a) if isinstance(a, Iterable) else value(a) for a in args)
|
||||||
out = func(*inputs) # type: ignore
|
out = func(*inputs)
|
||||||
tg.compile(out)
|
tg.compile(out)
|
||||||
_jit_cache[func] = (tg, inputs, out)
|
_jit_cache[func] = (tg, inputs, out)
|
||||||
tg.run()
|
tg.run()
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue