zebra/pyzebra
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix UB matrix read

Browse Source
This commit is contained in:
usov_i 2022-05-06 15:29:09 +02:00
parent cde2aa1182
commit 1d43c86cff
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pyzebra/ccl_io.py
View File

@ -1,5 +1,6 @@
import os import os
import re import re
from ast import literal_eval
from collections import defaultdict from collections import defaultdict
import numpy as np import numpy as np
@ -120,7 +121,7 @@ def parse_1D(fileobj, data_type):
elif variable in META_UB_MATRIX: elif variable in META_UB_MATRIX:
if variable == "UB": if variable == "UB":
metadata["ub"] = np.array(value) metadata["ub"] = np.array(literal_eval(value)).reshape(3, 3)
else: else:
if "ub" not in metadata: if "ub" not in metadata:
metadata["ub"] = np.zeros((3, 3)) metadata["ub"] = np.zeros((3, 3))