Skip to content

Commit b678f20

Browse files
author
DvirDukhan
committed
join on process
1 parent c495920 commit b678f20

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/flow/tests_tensorflow.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -656,6 +656,8 @@ def run(name=model_name, output_name='output{1}'):
656656
p1b = mp.Process(target=run, args=(another_model_name, 'final1{1}'))
657657
p1b.start()
658658

659+
p1b.join()
660+
659661
run(another_model_name, 'final2{1}')
660662

661663
_, dtype, _, shape, _, data = con.execute_command('AI.TENSORGET', 'final1{1}', 'META', 'BLOB')

0 commit comments

Comments
 (0)