Remove hardcoded version (#789)

Lokiiiiii · Jun 1, 2023 · 026e557 · 026e557
1 parent 663f389
commit 026e557
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/engines/python/setup/djl_python/np_util.py b/engines/python/setup/djl_python/np_util.py
@@ -144,7 +144,7 @@ def from_nd_list(encoded: bytearray) -> list:
             raise AssertionError("magic number is not NDAR, actual " + magic)
         version, idx = get_int(encoded, idx)
         if version != VERSION:
-            raise AssertionError("require version 2, actual " + str(version))
+            raise AssertionError(f"require version {VERSION}, actual " + str(version))
         flag, idx = get_byte_as_int(encoded, idx)
         if flag == 1:
             _, idx = get_str(encoded, idx)