From 14eb3565eb7f24c9c372ae3072bf04fca1e65e0f Mon Sep 17 00:00:00 2001
From: Markus <markushh@uio.no>
Date: Wed, 24 Sep 2025 20:23:53 +0200
Subject: [PATCH] fix: Duplicate predictions

---
 src/routines/output/predictions.rs | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/routines/output/predictions.rs b/src/routines/output/predictions.rs
index d6a961f2d..5e4b2529f 100644
--- a/src/routines/output/predictions.rs
+++ b/src/routines/output/predictions.rs
@@ -204,9 +204,10 @@ impl NPPredictions {
                 posterior_median.push(median_val);
             }
 
-            for pred in predictions.iter().enumerate() {
-                let (_, preds) = pred;
-                for (j, p) in preds.iter().enumerate() {
+            // Iterate over the aggregated predictions (one row per timepoint per subject)
+            // Use the first support point predictions to get time, outeq, block, and obs info
+            if let Some(first_spp_preds) = predictions.first() {
+                for (j, p) in first_spp_preds.iter().enumerate() {
                     let row = NPPredictionRow {
                         id: subject.id().clone(),
                         time: p.time(),