Skip to content

Commit d3fe950

Browse files
committed
fix joint prior name order
1 parent fae16ea commit d3fe950

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

bilby/core/prior/dict.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -885,14 +885,16 @@ def rescale(self, keys, theta):
885885
if isinstance(self[key], JointPrior):
886886
# if joint prior, keep track if all names have been rescaled
887887
distname = self[key].dist.distname
888-
names = set(self[key].dist.names)
888+
# maintain order of names as in the dist as this is the order
889+
# in which they will be rescaled
890+
names = self[key].dist.names
889891
if distname not in joint:
890892
joint[distname] = {key}
891893
else:
892894
joint[distname].add(key)
893895
# only when all names have been rescaled, we can set the values
894896
# we use sets because the order does not matter here
895-
if names == joint[distname]:
897+
if set(names) == joint[distname]:
896898
for name, value in zip(names, result[key]):
897899
result[name] = value
898900
self[name].least_recently_sampled = value

0 commit comments

Comments
 (0)