diff --git a/arraycontext/impl/pytato/compile.py b/arraycontext/impl/pytato/compile.py index 302dc659f2c34cb18931f5c98acf9f18970a0302..9fcd966158532dfcba4feb437c484d8f399978ee 100644 --- a/arraycontext/impl/pytato/compile.py +++ b/arraycontext/impl/pytato/compile.py @@ -106,11 +106,7 @@ def _get_arg_id_to_arg_and_arg_id_to_descr(args: Tuple[Any, ...] arg_id_to_arg: Dict[Tuple[Any, ...], Any] = {} arg_id_to_descr: Dict[Tuple[Any, ...], AbstractInputDescriptor] = {} - def to_arg_descr(iarg: int, arg: Any) -> None: - """ - Returns a :class:`AbstractInputDescriptor` for a - attr:`LazilyCompilingFunctionCaller.f`'s input argument. - """ + for iarg, arg in enumerate(args): if np.isscalar(arg): arg_id = (iarg,) arg_id_to_arg[arg_id] = arg @@ -129,9 +125,6 @@ def _get_arg_id_to_arg_and_arg_id_to_descr(args: Tuple[Any, ...] " either a scalar or an array container. Got" f" '{arg}'.") - for iarg, arg in enumerate(args): - to_arg_descr(iarg, arg) - return pmap(arg_id_to_arg), pmap(arg_id_to_descr)