replace pyrsistent.PMap, immutables.Map, immutabledict with constantdict

loopy/kernel/function_interface.py

@@ -348,6 +348,8 @@ def __init__(self,
         try:
             hash(arg_id_to_dtype)
         except TypeError:
+            if arg_id_to_dtype is None:
+                arg_id_to_dtype = {}
             arg_id_to_dtype = constantdict(arg_id_to_dtype)
             warn("arg_id_to_dtype passed to InKernelCallable was not hashable. "
                  "This usage is deprecated and will stop working in 2026.",
@@ -356,6 +358,8 @@ def __init__(self,
         try:
             hash(arg_id_to_descr)
         except TypeError:
+            if arg_id_to_descr is None:
+                arg_id_to_descr = {}
             arg_id_to_descr = constantdict(arg_id_to_descr)
             warn("arg_id_to_descr passed to InKernelCallable was not hashable. "
                  "This usage is deprecated and will stop working in 2026.",

loopy/schedule/tree.py

@@ -40,7 +40,7 @@
 from functools import cached_property, reduce
 from typing import Generic, TypeVar
 
-from constantdict import constantdict as Map
+from constantdict import constantdict as Map  # noqa: N812
 
 from pytools import memoize_method
 

loopy/translation_unit.py

@@ -302,14 +302,14 @@ def with_kernel(self, kernel: LoopKernel) -> Self:
             # update the callable kernel
             new_in_knl_callable = self.callables_table[kernel.name].copy(
                     subkernel=kernel)
-            new_callables = self.callables_table.delete(kernel.name).set(
-                    kernel.name, new_in_knl_callable)
-            return self.copy(callables_table=new_callables)
+            new_callables = self.callables_table.mutate()
+            new_callables[kernel.name] = new_in_knl_callable
+            return self.copy(callables_table=new_callables.finish())
         else:
             # add a new callable kernel
             clbl = CallableKernel(kernel)
-            new_callables = self.callables_table.set(kernel.name, clbl)
-            return self.copy(callables_table=new_callables)
+            return self.copy(
+                callables_table=self.callables_table.set(kernel.name, clbl))
 
     def __getitem__(self, name) -> LoopKernel:
         """