From 716f418ee3c985eea20d52e7146696f3250c53ff Mon Sep 17 00:00:00 2001 From: ggmarshall Date: Mon, 13 Nov 2023 13:04:08 +0100 Subject: [PATCH] check sig_out is array, needed for loading multiple files --- src/lgdo/compression/varlen.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/lgdo/compression/varlen.py b/src/lgdo/compression/varlen.py index f273e038..fb0f1163 100644 --- a/src/lgdo/compression/varlen.py +++ b/src/lgdo/compression/varlen.py @@ -227,7 +227,8 @@ def decode( # convert vector of vectors to array of equal sized arrays # can now decode on the 2D matrix together with number of bytes to read per row _, siglen = decode( - (sig_in.encoded_data.to_aoesa(preserve_dtype=True).nda, nbytes), sig_out.nda + (sig_in.encoded_data.to_aoesa(preserve_dtype=True).nda, nbytes), + sig_out if isinstance(sig_out, np.ndarray) else sig_out.nda, ) # sanity check