From 8d98eba403109cbf9341439b5a8941f776a5f007 Mon Sep 17 00:00:00 2001 From: Sander Roosendaal Date: Sun, 14 Apr 2024 20:52:40 +0200 Subject: [PATCH] fixing performance chart (polars) --- rowers/dataprep.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/rowers/dataprep.py b/rowers/dataprep.py index 42efff8f..c6f8049d 100644 --- a/rowers/dataprep.py +++ b/rowers/dataprep.py @@ -124,6 +124,10 @@ from rq import Queue from rowers.datautils import rpetotss def rscore_approx(row): + if isinstance(row, pl.DataFrame): + row = {'rscore': row['rscore'][0]} + if isinstance(row, pl.Series): + row = {'rscore': row['rscore'][0]} if row['rscore'] > 0: return row['rscore'] if row['rscore'] == 0: # pragma: no cover