From 03d5f8a5ab2ae43bcce7fd641a87eabf7f6585d9 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 6 Oct 2023 09:27:04 -0500
Subject: [PATCH] Replace pyrsistent.pmap with immutabledict

---
 grudge/array_context.py | 16 +++++++---------
 setup.py                |  2 +-
 2 files changed, 8 insertions(+), 10 deletions(-)

diff --git a/grudge/array_context.py b/grudge/array_context.py
index 3a7c0531..c5672178 100644
--- a/grudge/array_context.py
+++ b/grudge/array_context.py
@@ -39,7 +39,6 @@
 from meshmode.array_context import (
         PyOpenCLArrayContext as _PyOpenCLArrayContextBase,
         PytatoPyOpenCLArrayContext as _PytatoPyOpenCLArrayContextBase)
-from pyrsistent import pmap
 from warnings import warn
 
 import logging
@@ -275,8 +274,8 @@ def _dag_to_compiled_func(self, dict_of_named_arrays,
         # }}}
 
         part_id_to_prg = {}
-        name_in_program_to_tags = pmap()
-        name_in_program_to_axes = pmap()
+        name_in_program_to_tags = {}
+        name_in_program_to_axes = {}
 
         from pytato import make_dict_of_named_arrays
         for part in distributed_partition.parts.values():
@@ -295,19 +294,18 @@ def _dag_to_compiled_func(self, dict_of_named_arrays,
                         & set(part_prg_name_to_tags.keys()))
             assert not (set(name_in_program_to_axes.keys())
                         & set(part_prg_name_to_axes.keys()))
-            name_in_program_to_tags = name_in_program_to_tags.update(
-                part_prg_name_to_tags)
-            name_in_program_to_axes = name_in_program_to_axes.update(
-                part_prg_name_to_axes)
+            name_in_program_to_tags.update(part_prg_name_to_tags)
+            name_in_program_to_axes.update(part_prg_name_to_axes)
 
+        from immutabledict import immutabledict
         return _DistributedCompiledFunction(
                 actx=self.actx,
                 distributed_partition=distributed_partition,
                 part_id_to_prg=part_id_to_prg,
                 input_id_to_name_in_program=input_id_to_name_in_program,
                 output_id_to_name_in_program=output_id_to_name_in_program,
-                name_in_program_to_tags=name_in_program_to_tags,
-                name_in_program_to_axes=name_in_program_to_axes,
+                name_in_program_to_tags=immutabledict(name_in_program_to_tags),
+                name_in_program_to_axes=immutabledict(name_in_program_to_axes),
                 output_template=output_template)
 
 
diff --git a/setup.py b/setup.py
index adcbd6c0..f2fcad2d 100644
--- a/setup.py
+++ b/setup.py
@@ -49,7 +49,7 @@ def main():
             "pymbolic>=2013.2",
             "loopy>=2020.2",
             "cgen>=2013.1.2",
-            "dataclasses>=0.7;python_version<='3.6'"
+            "immutabledict",
         ],
     )