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 diff --git a/rowers/tests/testdata/testdata.tcx.gz b/rowers/tests/testdata/testdata.tcx.gz index 7cfb8e65..5735f761 100644 Binary files a/rowers/tests/testdata/testdata.tcx.gz and b/rowers/tests/testdata/testdata.tcx.gz differ