change Map -> immutabledict in outlining

Author: majosm

arraycontext/impl/pytato/outline.py

@@ -30,7 +30,7 @@
 from typing import Any, Callable, Dict, FrozenSet, Mapping, Optional, Tuple, Union
 
 import numpy as np
-from immutables import Map
+from immutabledict import immutabledict
 
 import pytato as pt
 from pytools.tag import Tag
@@ -43,7 +43,7 @@
 
 def _get_arg_id_to_arg(args: Tuple[Any, ...],
                        kwargs: Mapping[str, Any]
-                       ) -> Map[Tuple[Any, ...], Any]:
+                       ) -> immutabledict[Tuple[Any, ...], Any]:
     """
     Helper for :meth:`OulinedCall.__call__`. Extracts mappings from argument id
     to argument values. See
@@ -77,7 +77,7 @@ def id_collector(keys, ary):
                              " either a scalar, pt.Array or an array container. Got"
                              f" '{arg}'.")
 
-    return Map(arg_id_to_arg)
+    return immutabledict(arg_id_to_arg)
 
 
 def _get_input_arg_id_str(
@@ -95,14 +95,15 @@ def _get_output_arg_id_str(arg_id: Tuple[Any, ...]) -> str:
 
 def _get_arg_id_to_placeholder(
         arg_id_to_arg: Mapping[Tuple[Any, ...], Any],
-        prefix: Optional[str] = None) -> Map[Tuple[Any, ...], pt.Placeholder]:
+        prefix: Optional[str] = None
+    ) -> immutabledict[Tuple[Any, ...], pt.Placeholder]:
     """
     Helper for :meth:`OulinedCall.__call__`. Constructs a :class:`pytato.Placeholder`
     for each argument in *arg_id_to_arg*. See
     :attr:`CompiledFunction.input_id_to_name_in_function` for argument-id's
     representation.
     """
-    return Map({
+    return immutabledict({
         arg_id: pt.make_placeholder(
             _get_input_arg_id_str(arg_id, prefix=prefix),
             arg.shape,
@@ -244,7 +245,7 @@ def __call__(self, *args: Any, **kwargs: Any) -> ArrayOrContainer:
         func_def = pt.function.FunctionDefinition(
             parameters=frozenset(call_bindings.keys()),
             return_type=ret_type,
-            returns=Map(unpacked_output),
+            returns=immutabledict(unpacked_output),
             tags=self.tags,
         )