From 208ed3c52d909ba1a896649564575b37bdad08b3 Mon Sep 17 00:00:00 2001 From: Kaushik Kulkarni Date: Sun, 20 Mar 2022 11:25:45 -0500 Subject: [PATCH] materialize nodes before find_distributed_partition --- test/test_distributed.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/test_distributed.py b/test/test_distributed.py index 3ac44ad..0e67911 100644 --- a/test/test_distributed.py +++ b/test/test_distributed.py @@ -166,10 +166,10 @@ def _do_test_distributed_execution_random_dag(ctx_factory): additional_generators=[ (comm_fake_prob, gen_comm) ]) - x_comm = make_random_dag(rdagc_comm) + pt_dag = pt.DictOfNamedArrays({"result": make_random_dag(rdagc_comm)}) + x_comm = pt.transform.materialize_with_mpms(pt_dag) - distributed_partition = find_distributed_partition( - pt.DictOfNamedArrays({"result": x_comm})) + distributed_partition = find_distributed_partition(x_comm) # Transform symbolic tags into numeric ones for MPI distributed_partition, _new_mpi_base_tag = number_distributed_tags( -- GitLab