From 026e557bc49969b337c3504e0a78e8501f029940 Mon Sep 17 00:00:00 2001 From: Sindhu Somasundaram <56774226+sindhuvahinis@users.noreply.github.com> Date: Thu, 1 Jun 2023 09:18:31 -0700 Subject: [PATCH] Remove hardcoded version (#789) --- engines/python/setup/djl_python/np_util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/engines/python/setup/djl_python/np_util.py b/engines/python/setup/djl_python/np_util.py index d2e8acd4a..d1c4f53f9 100644 --- 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)