Skip to content

Commit fbc0167

Browse files
author
DvirDukhan
committed
join threads on tests
1 parent 130ed18 commit fbc0167

File tree

3 files changed

+3
-0
lines changed

3 files changed

+3
-0
lines changed

tests/flow/tests_dag.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -771,6 +771,7 @@ def run():
771771
# assert that result tensor exists
772772
ret = con.execute_command("EXISTS {}".format(result_tensor_keyname))
773773
env.assertEqual(ret, 1)
774+
t.join()
774775

775776

776777
def test_dag_with_timeout(env):

tests/flow/tests_onnx.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -229,6 +229,7 @@ def run():
229229
argmax = max(range(len(values)), key=lambda i: values[i])
230230

231231
env.assertEqual(argmax, 1)
232+
t.join()
232233

233234

234235
def test_onnx_modelrun_iris(env):

tests/flow/tests_pytorch.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -284,6 +284,7 @@ def run():
284284
exception = e
285285
env.assertEqual(type(exception), redis.exceptions.ResponseError)
286286
env.assertEqual("Model did not generate the expected batch size", exception.__str__())
287+
t.join()
287288

288289

289290

0 commit comments

Comments
 (0)