From 9c1de25cb8cdcea6f56d19ddfd18cad51478fc83 Mon Sep 17 00:00:00 2001
From: Sander Roosendaal <sanderroosendaal@proton.me>
Date: Mon, 20 May 2024 15:07:08 +0200
Subject: [PATCH] fix

---
 rowers/dataprep.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/rowers/dataprep.py b/rowers/dataprep.py
index 620e40fe..0f4370a1 100644
--- a/rowers/dataprep.py
+++ b/rowers/dataprep.py
@@ -127,9 +127,21 @@ from rowers.datautils import rpetotss
 
 def rscore_approx(row):
     if isinstance(row, pl.DataFrame):
-        row = {'rscore': row['rscore'][0]}
+        row = {
+            'rscore': row['rscore'][0],
+            'hrtss': row['hrtss'][0],
+            'id': row['id'][0],
+            'rpe': row['rpe'][0],
+            'duration': row['duration'][0],
+        }
     if isinstance(row, pl.Series):
-        row = {'rscore': row['rscore'][0]}
+        row = {
+            'rscore': row['rscore'][0],
+            'hrtss': row['hrtss'][0],
+            'id': row['id'][0],
+            'rpe': row['rpe'][0],
+            'duration': row['duration'][0],
+        }
     if row['rscore'] > 0:
         return row['rscore']
     if row['rscore'] == 0: # pragma: no cover