diff --git a/features/calendar.py b/features/calendar.py index 9729caa..d2c37bd 100644 --- a/features/calendar.py +++ b/features/calendar.py @@ -71,11 +71,12 @@ class Calendar: 'sets': [] } - workouts_by_date[workout_date_str][workout_id]['sets'].append({ - 'repetitions': row['repetitions'], - 'weight': row['weight'], - 'exercise_name': row['exercise_name'] - }) + if row['topset_id']: + workouts_by_date[workout_date_str][workout_id]['sets'].append({ + 'repetitions': row['repetitions'], + 'weight': row['weight'], + 'exercise_name': row['exercise_name'] + }) for current_date in pd.date_range(start_date, end_date, freq='D'): date_str = current_date.strftime("%Y-%m-%d")