Skip to content

Commit 312cf3a

Browse files
committed
fix(levels): reduce force-torque outputs deterministically
1 parent 6ba1114 commit 312cf3a

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

CodeEntropy/levels/level_dag.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -315,13 +315,15 @@ def _reduce_forcetorque(
315315
gid2i = shared_data["group_id_to_index"]
316316
ft_frame = frame_out["forcetorque"]
317317

318-
for gid, M in ft_frame.get("res", {}).items():
318+
for gid in sorted(ft_frame.get("res", {}).keys()):
319+
M = ft_frame["res"][gid]
319320
gi = gid2i[gid]
320321
ft_counts["res"][gi] += 1
321322
n = ft_counts["res"][gi]
322323
ft_cov["res"][gi] = self._incremental_mean(ft_cov["res"][gi], M, n)
323324

324-
for gid, M in ft_frame.get("poly", {}).items():
325+
for gid in sorted(ft_frame.get("poly", {}).keys()):
326+
M = ft_frame["poly"][gid]
325327
gi = gid2i[gid]
326328
ft_counts["poly"][gi] += 1
327329
n = ft_counts["poly"][gi]

0 commit comments

Comments
 (0)