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