use immutabledict instead of pyrsistent, immutabels

meshmode/array_context.py

@@ -53,7 +53,7 @@
                                          DiscretizationEntityAxisTag)
 from dataclasses import dataclass
 
-from pyrsistent import pmap
+from immutabledict import immutabledict
 logger = logging.getLogger(__name__)
 
 
@@ -781,15 +781,14 @@ class NotAnFEMEinsumError(ValueError):
 
 @memoize_on_first_arg
 def _get_redn_iname_to_insns(kernel):
-    from immutables import Map
     redn_iname_to_insns = {iname: set()
                            for iname in kernel.all_inames()}
 
     for insn in kernel.instructions:
         for redn_iname in insn.reduction_inames():
             redn_iname_to_insns[redn_iname].add(insn.id)
 
-    return Map({k: frozenset(v)
+    return immutabledict({k: frozenset(v)
                 for k, v in redn_iname_to_insns.items()})
 
 
@@ -1072,7 +1071,7 @@ def _get_iel_to_idofs(kernel):
             raise NotImplementedError(f"Cannot fit loop nest '{insn.within_inames}'"
                                       " into known set of loop-nest patterns.")
 
-    return pmap({iel: frozenset(idofs)
+    return immutabledict({iel: frozenset(idofs)
                  for iel, idofs in iel_to_idofs.items()})