fix: summarize Rain metric trajectories

1 files changed, 1 insertions, 1 deletions

diff --git a/experiments/rain_ep_centered_r3.sh b/experiments/rain_ep_centered_r3.sh
index 2ea5acb..be44649 100755
--- a/experiments/rain_ep_centered_r3.sh
+++ b/experiments/rain_ep_centered_r3.sh
@@ -54,7 +54,7 @@ by_seed = {}
 for path in sorted(root.glob("seed_*.json")):
     _, seed, method = path.stem.split("_", 2)
     report = json.loads(path.read_text())
-    trajectory = [epoch["test_accuracy"] for epoch in report["epochs"]]
+    trajectory = [epoch["test_accuracy"] for epoch in report["metrics"]]
     by_seed.setdefault(seed, {})[method] = {
         "trajectory": trajectory,
         "best_accuracy": max(trajectory),