fix test

cunla · Jun 4, 2023 · afeae76 · afeae76
1 parent ccbeb13
commit afeae76
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/fakeredis/_commands.py b/fakeredis/_commands.py
@@ -6,6 +6,8 @@
 import math
 import re
 
+from packaging.version import Version
+
 from . import _msgs as msgs
 from ._helpers import null_terminate, SimpleError, SimpleString
 from ._zset import ZSet
@@ -330,7 +332,7 @@ def apply(self, args, db, version):
         if self.repeat:
             delta = len(args) - len(self.fixed)
             if delta % len(self.repeat) != 0:
-                msg = msgs.WRONG_ARGS_MSG7 if version >= 7 else msgs.WRONG_ARGS_MSG6.format(self.name)
+                msg = msgs.WRONG_ARGS_MSG7 if version >= Version('7') else msgs.WRONG_ARGS_MSG6.format(self.name)
                 raise SimpleError(msg)
 
         types = list(self.fixed)