Skip to content

Commit 76eba50

Browse files
committed
Minor shuffle version handling
1 parent bc68c28 commit 76eba50

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

python/shuffle.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -366,6 +366,8 @@ def get_numpy_npz_headers(filename):
366366
header = np.lib.format.read_array_header_1_0(npyfile)
367367
elif version == (2, 0):
368368
header = np.lib.format.read_array_header_2_0(npyfile)
369+
else:
370+
raise NotImplementedError(f"Unexpected np version for {filename}: {version}")
369371
(shape, is_fortran, dtype) = header
370372
npzheaders[subfilename] = (shape, is_fortran, dtype)
371373
if wasbad:

0 commit comments

Comments
 (0)