Skip to content
Snippets Groups Projects
Commit fcda4c82 authored by Kaushik Kulkarni's avatar Kaushik Kulkarni
Browse files

inline to_arg_descr

parent 5935dea8
No related branches found
No related tags found
No related merge requests found
...@@ -106,11 +106,7 @@ def _get_arg_id_to_arg_and_arg_id_to_descr(args: Tuple[Any, ...] ...@@ -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_arg: Dict[Tuple[Any, ...], Any] = {}
arg_id_to_descr: Dict[Tuple[Any, ...], AbstractInputDescriptor] = {} arg_id_to_descr: Dict[Tuple[Any, ...], AbstractInputDescriptor] = {}
def to_arg_descr(iarg: int, arg: Any) -> None: for iarg, arg in enumerate(args):
"""
Returns a :class:`AbstractInputDescriptor` for a
attr:`LazilyCompilingFunctionCaller.f`'s input argument.
"""
if np.isscalar(arg): if np.isscalar(arg):
arg_id = (iarg,) arg_id = (iarg,)
arg_id_to_arg[arg_id] = arg 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, ...] ...@@ -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" " either a scalar or an array container. Got"
f" '{arg}'.") f" '{arg}'.")
for iarg, arg in enumerate(args):
to_arg_descr(iarg, arg)
return pmap(arg_id_to_arg), pmap(arg_id_to_descr) return pmap(arg_id_to_arg), pmap(arg_id_to_descr)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment