diff --git a/grudge/array_context.py b/grudge/array_context.py index 2e82519e2..3a7c0531c 100644 --- a/grudge/array_context.py +++ b/grudge/array_context.py @@ -278,9 +278,9 @@ def _dag_to_compiled_func(self, dict_of_named_arrays, name_in_program_to_tags = pmap() name_in_program_to_axes = pmap() - from pytato import DictOfNamedArrays + from pytato import make_dict_of_named_arrays for part in distributed_partition.parts.values(): - d = DictOfNamedArrays( + d = make_dict_of_named_arrays( {var_name: distributed_partition.name_to_output[var_name] for var_name in part.output_names })