diff --git a/rowers/interactiveplots.py b/rowers/interactiveplots.py index 0e214a83..9c99d441 100644 --- a/rowers/interactiveplots.py +++ b/rowers/interactiveplots.py @@ -278,7 +278,12 @@ def interactive_workouttype_piechart(workouts): data = pd.Series(datadict).reset_index(name='value').rename(columns={'index':'type'}) data['angle'] = data['value']/data['value'].sum() * 2*pi - data['color'] = Category20c[len(datadict)] + if len(datadict)>2: + data['color'] = Category20c[len(datadict)] + elif len(datadict)==2: + data['color'] = ['orange','lightblue'] + else: + data['color'] = ['lightblue'] data['totaltime'] = pd.Series([pretty_timedelta(v) for v in data['value']]) p = figure(plot_height=350, title="Types", toolbar_location=None,