type checker annotation fixed for jit decorator function

This commit is contained in:
Nicolas Kruse 2025-12-19 20:43:01 +01:00
parent ce15e83c2b
commit a07332f3a4
1 changed files with 1 additions and 1 deletions

View File

@ -32,7 +32,7 @@ def jit(func: Callable[..., TRet]) -> Callable[..., TRet]:
tg = Target()
inputs = tuple(
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)
_jit_cache[func] = (tg, inputs, out)
tg.run()