From e8b5806732bdbf3f415f53df71a20b109cda10cd Mon Sep 17 00:00:00 2001 From: Matthias Diener Date: Mon, 14 Oct 2024 11:46:22 -0500 Subject: [PATCH] fix materialized_arrays perf --- pytato/distributed/partition.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytato/distributed/partition.py b/pytato/distributed/partition.py index 111f07d2e..cf75e6ed7 100644 --- a/pytato/distributed/partition.py +++ b/pytato/distributed/partition.py @@ -844,7 +844,7 @@ def find_distributed_partition( # processing below. materialized_arrays = {a: None for a in materialized_arrays_collector.materialized_arrays - if a not in received_arrays | sent_arrays} + if a not in received_arrays and a not in sent_arrays} # "mso" for "materialized/sent/output" output_arrays = dict.fromkeys(outputs._data.values())