Merge pull request #46 from RedisAI/no-load-persist-dag

Fix DAGRUN with no LOAD or PERSIST

Author: Sherin Thomas

redisai/client.py

@@ -632,7 +632,7 @@ def __init__(self, load, persist, executor, readonly=False, postprocess=True):
                 self.commands += ["PERSIST", 1, persist, '|>']
             else:
                 self.commands += ["PERSIST", len(persist), *persist, '|>']
-        elif load:
+        else:
             self.commands.append('|>')
         self.executor = executor
 

test/test.py

@@ -334,8 +334,9 @@ def test_dagrun_with_load(self):
     def test_dagrun_with_persist(self):
         con = self.get_client()
 
-        dag = con.dag(persist='wrongkey')  # this won't raise Error
-        dag.tensorset('a', [2, 3, 2, 3], shape=(2, 2), dtype='float').run()
+        with self.assertRaises(ResponseError):
+            dag = con.dag(persist='wrongkey')
+            dag.tensorset('a', [2, 3, 2, 3], shape=(2, 2), dtype='float').run()
 
         dag = con.dag(persist=['b'])
         dag.tensorset('a', [2, 3, 2, 3], shape=(2, 2), dtype='float')