Cast f[indices] to a list

parent 258a4918
......@@ -252,7 +252,7 @@ class VideoLikeContainer:
# weak closing of the hdf5 file so we don't load all the data into
# memory https://docs.h5py.org/en/stable/high/file.html#closing-files
f = h5py.File(file, mode="r")
loaded = {"data": f["data"], "indices": f["indices"]}
loaded = {"data": f["data"], "indices": list(f["indices"])}
except OSError:
with open(file, "rb") as f:
loaded = pickle.load(f)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment