fixed PR comments

Author: DvirDukhan

tests/flow/tests_dag.py

@@ -761,6 +761,7 @@ def run():
                 'AI.TENSORGET', result_tensor_keyname, 'META',
             )
 
+            t.join()
             ensureSlaveSynced(con, env)
 
             env.assertEqual([b'OK',b'OK',[b'dtype', b'FLOAT', b'shape', [1, 2]]], ret)
@@ -771,7 +772,6 @@ def run():
             # assert that result tensor exists
             ret = con.execute_command("EXISTS {}".format(result_tensor_keyname))
             env.assertEqual(ret, 1)
-            t.join()
 
 
 def test_dag_with_timeout(env):

tests/flow/tests_onnx.py

@@ -214,6 +214,7 @@ def run():
     t.start()
 
     con.execute_command('AI.MODELRUN', 'm{1}', 'INPUTS', 'a{1}', 'OUTPUTS', 'b{1}')
+    t.join()
 
     ensureSlaveSynced(con, env)
 
@@ -229,7 +230,6 @@ def run():
     argmax = max(range(len(values)), key=lambda i: values[i])
 
     env.assertEqual(argmax, 1)
-    t.join()
 
 
 def test_onnx_modelrun_iris(env):

tests/flow/tests_pytorch.py

@@ -269,7 +269,8 @@ def test_pytorch_modelrun_autobatch_badbatch(env):
     def run():
         con = env.getConnection()
         try:
-            con.execute_command('AI.MODELRUN', 'm{1}', 'INPUTS', 'd{1}', 'e{1}', 'OUTPUTS', 'f1{1}', 'f2{1}')
+            ret = con.execute_command('AI.MODELRUN', 'm{1}', 'INPUTS', 'd{1}', 'e{1}', 'OUTPUTS', 'f1{1}', 'f2{1}')
+            env.assertEqual(ret, b'OK')
         except Exception as e:
             exception = e
             env.assertEqual(type(exception), redis.exceptions.ResponseError)
@@ -279,7 +280,8 @@ def run():
     t.start()
 
     try:
-        con.execute_command('AI.MODELRUN', 'm{1}', 'INPUTS', 'a{1}', 'b{1}', 'OUTPUTS', 'c1{1}', 'c2{1}')
+        ret = con.execute_command('AI.MODELRUN', 'm{1}', 'INPUTS', 'a{1}', 'b{1}', 'OUTPUTS', 'c1{1}', 'c2{1}')
+        env.assertEqual(ret, b'OK')
     except Exception as e:
         exception = e
         env.assertEqual(type(exception), redis.exceptions.ResponseError)