From 07b83fce9117bc68f2e56f0761c7bf917a75404f Mon Sep 17 00:00:00 2001 From: Guillaume Viejo Date: Wed, 5 Feb 2025 10:46:15 -0500 Subject: [PATCH] quick fix --- pynapple/core/ts_group.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/pynapple/core/ts_group.py b/pynapple/core/ts_group.py index 3709f103..75bd6cc3 100644 --- a/pynapple/core/ts_group.py +++ b/pynapple/core/ts_group.py @@ -1419,13 +1419,21 @@ def _from_npz_reader(cls, file): tsgroup = cls(group, time_support=time_support, bypass_check=True) - # do we need to enforce that these keys are not in metadata? - # not_info_keys = {"start", "end", "t", "index", "d", "rate", "keys"} - if "_metadata" in file: # load metadata if it exists if file["_metadata"]: # check that metadata is not empty metainfo = pd.DataFrame.from_dict(file["_metadata"].item()) tsgroup.set_info(metainfo) + + metainfo = {} + not_info_keys = {"start", "end", "t", "index", "d", "rate", "keys", "_metadata"} + + for k in set(file.keys()) - not_info_keys: + tmp = file[k] + if len(tmp) == len(tsgroup): + metainfo[k] = tmp + + tsgroup.set_info(**metainfo) + return tsgroup @add_meta_docstring("set_info")