From c9a718d03f5b2f1caacd1bbade5169629e32ff94 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Al=C3=A1n=20Mu=C3=B1oz?= <amuoz@ed.ac.uk>
Date: Wed, 3 Aug 2022 17:46:17 +0100
Subject: [PATCH] feat(load_attributes): add parameters as dict
---
io/writer.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/io/writer.py b/io/writer.py
index 1a59358f..d3030bfd 100644
--- a/io/writer.py
+++ b/io/writer.py
@@ -7,6 +7,7 @@ from typing import Dict
import h5py
import numpy as np
import pandas as pd
+import yaml
from utils_find_1st import cmp_equal, find_1st
from agora.io.bridge import BridgeH5
@@ -18,6 +19,8 @@ from agora.io.utils import timed
def load_attributes(file: str, group="/"):
with h5py.File(file, "r") as f:
meta = dict(f[group].attrs.items())
+ if "parameters" in meta:
+ meta["parameters"] = yaml.safe_load(meta["parameters"])
return meta
--
GitLab