From fcda4c825d664dacac019cbcbed4d8ba6d8c913e Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Tue, 29 Jun 2021 06:10:04 -0500
Subject: [PATCH] inline to_arg_descr
---
arraycontext/impl/pytato/compile.py | 9 +--------
1 file changed, 1 insertion(+), 8 deletions(-)
diff --git a/arraycontext/impl/pytato/compile.py b/arraycontext/impl/pytato/compile.py
index 302dc65..9fcd966 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)
--
GitLab