Remove hardcoded version (deepjavalibrary#789)

KexinFeng · Aug 16, 2023 · f2d7808 · f2d7808
1 parent 1f7e9bd
commit f2d7808
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)