diff --git a/one/alf/io.py b/one/alf/io.py index 37a3cec4baebee64aca0abc48ab3276a484d7299..e62919eb9b71dbffc563dcd54526e676cda36d66 100644 --- a/one/alf/io.py +++ b/one/alf/io.py @@ -280,6 +280,8 @@ def load_file_content(fil): return jsonable.read(fil) if fil.suffix == '.npy': return _ensure_flat(np.load(file=fil, allow_pickle=True)) + if fil.suffix == '.npz': + return _ensure_flat(np.load(file=fil, allow_pickle=True)['arr_0']) if fil.suffix == '.pqt': return parquet.load(fil)[0] if fil.suffix == '.ssv':