From b5537af7834b4662cf046f43426439fde1cdfac4 Mon Sep 17 00:00:00 2001 From: Sander Roosendaal Date: Mon, 12 Jul 2021 08:05:11 +0200 Subject: [PATCH] bug fixes --- rowers/sporttracksstuff.py | 5 ++++- rowers/utils.py | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/rowers/sporttracksstuff.py b/rowers/sporttracksstuff.py index 94b99ce2..795d6067 100644 --- a/rowers/sporttracksstuff.py +++ b/rowers/sporttracksstuff.py @@ -157,7 +157,10 @@ def createsporttracksworkoutdata(w): # adding diff, trying to see if this is valid #t = row.df.loc[:,'TimeStamp (sec)'].values-10*row.df.ix[0,'TimeStamp (sec)'] t = row.df.loc[:,'TimeStamp (sec)'].values-row.df.loc[:,'TimeStamp (sec)'].iloc[0] - t[0] = t[1] + try: + t[0] = t[1] + except IndexError: + return 0 d = row.df.loc[:,'cum_dist'].values d[0] = d[1] diff --git a/rowers/utils.py b/rowers/utils.py index 3d26097c..335617b2 100644 --- a/rowers/utils.py +++ b/rowers/utils.py @@ -388,7 +388,10 @@ def wavg(group, avg_name, weight_name): In rare instance, we may not have weights, so just return the mean. Customize this if your business case should return otherwise. """ - d = group[avg_name] + try: + d = group[avg_name] + except KeyError: + return 0 try: w = group[weight_name] except KeyError: