diff --git a/dace/symbolic.py b/dace/symbolic.py index 5bfda9810c..d2f14da09d 100644 --- a/dace/symbolic.py +++ b/dace/symbolic.py @@ -1376,7 +1376,8 @@ def _print_Float(self, expr): def _print_Function(self, expr): if str(expr.func) in self.arrays: - return f'{expr.func}[{expr.args[0]}]' + indices = ", ".join(self._print(arg) for arg in expr.args) + return f'{expr.func}[{indices}]' if self.cpp_mode and str(expr.func) == 'int_floor': return '((%s) / (%s))' % (self._print(expr.args[0]), self._print(expr.args[1])) if str(expr.func) == 'AND':