sander/rowsandall
Private
Public Access
1
0
Fork 0
Code Activity

fixing performance chart (polars)

Browse Source
This commit is contained in:
Sander Roosendaal
2024-04-14 20:52:40 +02:00
parent 2d2923362f
commit 8d98eba403
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rowers/dataprep.py
View File

@@ -124,6 +124,10 @@ from rq import Queue
from rowers.datautils import rpetotss from rowers.datautils import rpetotss
def rscore_approx(row): 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: if row['rscore'] > 0:
return row['rscore'] return row['rscore']
if row['rscore'] == 0: # pragma: no cover if row['rscore'] == 0: # pragma: no cover