diff --git a/pysaliency/datasets/utils.py b/pysaliency/datasets/utils.py index 2fc89c9..6bf30ce 100644 --- a/pysaliency/datasets/utils.py +++ b/pysaliency/datasets/utils.py @@ -94,7 +94,7 @@ def _load_attribute_dict_from_hdf5(attribute_group): json_attributes = json_attributes.decode('utf8') __attributes__ = json.loads(json_attributes) - attributes = {attribute: load_hdf5_dataset(attribute_group, attribute) for attribute in __attributes__} + attributes = {attribute: attribute_group[attribute][...] for attribute in __attributes__} return attributes @@ -122,4 +122,4 @@ def concatenate_attributes(attributes): padded_attributes.append(np.hstack((a, padding))) else: padded_attributes.append(a) - return np.vstack(padded_attributes) \ No newline at end of file + return np.vstack(padded_attributes)