Skip to content
Snippets Groups Projects
Commit c9a718d0 authored by Alán Muñoz's avatar Alán Muñoz
Browse files

feat(load_attributes): add parameters as dict

parent f64117ce
No related branches found
No related tags found
No related merge requests found
...@@ -7,6 +7,7 @@ from typing import Dict ...@@ -7,6 +7,7 @@ from typing import Dict
import h5py import h5py
import numpy as np import numpy as np
import pandas as pd import pandas as pd
import yaml
from utils_find_1st import cmp_equal, find_1st from utils_find_1st import cmp_equal, find_1st
from agora.io.bridge import BridgeH5 from agora.io.bridge import BridgeH5
...@@ -18,6 +19,8 @@ from agora.io.utils import timed ...@@ -18,6 +19,8 @@ from agora.io.utils import timed
def load_attributes(file: str, group="/"): def load_attributes(file: str, group="/"):
with h5py.File(file, "r") as f: with h5py.File(file, "r") as f:
meta = dict(f[group].attrs.items()) meta = dict(f[group].attrs.items())
if "parameters" in meta:
meta["parameters"] = yaml.safe_load(meta["parameters"])
return meta return meta
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment