Skip to content

Commit bf9b404

Browse files
committed
fix for LPS exercise 2
1 parent 485e60a commit bf9b404

File tree

1 file changed

+16
-4
lines changed

1 file changed

+16
-4
lines changed

obsarray/unc_accessor.py

Lines changed: 16 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -308,12 +308,24 @@ def err_corr_dict(self) -> dict:
308308
]
309309
):
310310
if dim_err_corr[1].form in ["random", "systematic"]:
311-
err_corr_dict[dim_err_corr[0]] = dim_err_corr[1].form
311+
if isinstance(dim_err_corr[0], List):
312+
for dim in dim_err_corr[0]:
313+
err_corr_dict[dim] = dim_err_corr[1].form
314+
315+
else:
316+
err_corr_dict[dim_err_corr[0]] = dim_err_corr[1].form
312317

313318
elif dim_err_corr[1].form == "err_corr_matrix":
314-
err_corr_dict[dim_err_corr[0]] = self._obj[
315-
dim_err_corr[1].params[0]
316-
].values
319+
if isinstance(dim_err_corr[0], List):
320+
for dim in dim_err_corr[0]:
321+
err_corr_dict[dim] = self._obj[
322+
dim_err_corr[1].params[0]
323+
].values
324+
325+
else:
326+
err_corr_dict[dim_err_corr[0]] = self._obj[
327+
dim_err_corr[1].params[0]
328+
].values
317329

318330
else:
319331
raise NotImplementedError(

0 commit comments

Comments
 (0)