diff --git a/arraycontext/impl/pytato.py b/arraycontext/impl/pytato.py index 739a92231300e18ac16581d17994e95c39f88183..9636448dbb294c19b0755fa3bb5a48ed1b4ea25b 100644 --- a/arraycontext/impl/pytato.py +++ b/arraycontext/impl/pytato.py @@ -193,7 +193,7 @@ class ArrayContainerInputDescriptor(AbstractInputDescriptor): @dataclass class PytatoCompiledOperator: actx: ArrayContext - f: Callable[[Any], Any] + f: Callable[[...], Any] program_cache: Mapping[Tuple[AbstractInputDescriptor], "PytatoExecutable"] = field(default_factory=lambda: {})