Skip to content

Commit 27ae16c

Browse files
committed
Fix test_multi_process_transparent_api_create_on_local_remote_workflow
1 parent ac653d8 commit 27ae16c

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

tests/slow/test_remoteworkflow.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -552,25 +552,28 @@ def test_multi_process_transparent_api_connect_local_op_remote_workflow():
552552
)
553553
def test_multi_process_transparent_api_create_on_local_remote_workflow():
554554
files = examples.download_distributed_files()
555-
wf = core.Workflow()
555+
# Make sure to reuse the same type of remote server as for the previous ones:
556+
# Cannot merge a Workflow from a non-legacy grpc server to a workflow on a legacy grpc server
557+
merge_server = local_servers[len(files)]
558+
wf = core.Workflow(server=merge_server)
556559
wf.progress_bar = False
557-
op = ops.result.displacement()
558-
average = core.operators.math.norm_fc(op)
560+
op = ops.result.displacement(server=merge_server)
561+
average = core.operators.math.norm_fc(op, server=merge_server)
559562

560563
wf.add_operators([op, average])
561564
wf.set_output_name("distrib", average.outputs.fields_container)
562565
wf.set_input_name("ds", op.inputs.data_sources)
563566

564-
local_wf = core.Workflow()
567+
local_wf = core.Workflow(server=merge_server)
565568
local_wf.progress_bar = False
566-
merge = ops.utility.merge_fields_containers()
567-
min_max = ops.min_max.min_max_fc(merge)
569+
merge = ops.utility.merge_fields_containers(server=merge_server)
570+
min_max = ops.min_max.min_max_fc(merge, server=merge_server)
568571
local_wf.add_operator(merge)
569572
local_wf.add_operator(min_max)
570573
local_wf.set_output_name("tot_output", min_max.outputs.field_max)
571574

572575
for i in files:
573-
data_sources1 = core.DataSources(files[i])
576+
data_sources1 = core.DataSources(files[i], server=merge_server)
574577
remote_wf = wf.create_on_other_server(server=local_servers[i])
575578
remote_wf.connect("ds", data_sources1)
576579
local_wf.set_input_name("distrib" + str(i), merge, i)

0 commit comments

Comments
 (0)