After merging a fix to RLTest - this is for covering aof rewrite in tests + fixing minor bug in model store (#800)

Author: alonre24

src/redisai.c

@@ -208,23 +208,23 @@ int RedisAI_ModelStore_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **arg
 
     unsigned long long minbatchsize = 0;
     if (AC_AdvanceIfMatch(&ac, "MINBATCHSIZE")) {
-        if (batchsize == 0) {
-            return RedisModule_ReplyWithError(ctx, "ERR MINBATCHSIZE specified without BATCHSIZE");
-        }
         if (AC_GetUnsignedLongLong(&ac, &minbatchsize, 0) != AC_OK) {
             return RedisModule_ReplyWithError(ctx, "ERR Invalid argument for MINBATCHSIZE");
         }
+        if (batchsize == 0 && minbatchsize > 0) {
+            return RedisModule_ReplyWithError(ctx, "ERR MINBATCHSIZE specified without BATCHSIZE");
+        }
     }
 
     unsigned long long minbatchtimeout = 0;
     if (AC_AdvanceIfMatch(&ac, "MINBATCHTIMEOUT")) {
-        if (minbatchsize == 0) {
-            return RedisModule_ReplyWithError(ctx,
-                                              "ERR MINBATCHTIMEOUT specified without MINBATCHSIZE");
-        }
         if (AC_GetUnsignedLongLong(&ac, &minbatchtimeout, 0) != AC_OK) {
             return RedisModule_ReplyWithError(ctx, "ERR Invalid argument for MINBATCHTIMEOUT");
         }
+        if (minbatchsize == 0 && minbatchtimeout > 0) {
+            return RedisModule_ReplyWithError(ctx,
+                                              "ERR MINBATCHTIMEOUT specified without MINBATCHSIZE");
+        }
     }
     RAI_ModelOpts opts = {
         .batchsize = batchsize,

tests/flow/test_serializations.py

@@ -276,7 +276,7 @@ def test_v3_tensor(self):
 class TestAofRewrite:
 
     def __init__(self):
-        self.env = Env(useAof=True)
+        self.env = Env(useAof=True, useRdbPreamble=False)
 
     def test_aof_rewrite_tf_model(self):
         key_name = "tf_graph{1}"