Dict -> Mapping

inducer · Feb 10, 2024 · 3bdad69 · 3bdad69
1 parent ae41332
commit 3bdad69
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pytato/transform/__init__.py b/pytato/transform/__init__.py
@@ -474,9 +474,9 @@ def rec_idx_or_size_tuple(self, situp: Tuple[IndexOrShapeExpr, ...],
 
     def map_index_lambda(self, expr: IndexLambda,
                          *args: Any, **kwargs: Any) -> Array:
-        bindings: Dict[str, Array] = {
-                name: self.rec(subexpr, *args, **kwargs)
-                for name, subexpr in sorted(expr.bindings.items())}
+        bindings: Mapping[str, Array] = immutabledict({
+                name: self.rec(subexpr)
+                for name, subexpr in sorted(expr.bindings.items())})
         return IndexLambda(expr=expr.expr,
                            shape=self.rec_idx_or_size_tuple(expr.shape,
                                                             *args, **kwargs),