fix issue in PlaceholderSubstitutor call mapping

inducer · Aug 27, 2024 · a5dbc5a · a5dbc5a
1 parent 97212b9
commit a5dbc5a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytato/transform/calls.py b/pytato/transform/calls.py
@@ -70,7 +70,8 @@ def map_placeholder(self, expr: Placeholder) -> Array:
         # collide with the ones being replaced
         return self.substitutions[expr.name]
 
-    def map_call(self, expr: Call) -> Call:
+    def map_function_definition(
+            self, expr: FunctionDefinition) -> FunctionDefinition:
         # Only operates within the current stack frame
         return expr