diff --git a/fairml/graphing.py b/fairml/graphing.py index e20d384..1e02bfc 100644 --- a/fairml/graphing.py +++ b/fairml/graphing.py @@ -61,7 +61,7 @@ def map_x(x): fig = plt.figure(figsize=fig_size) - bar_colors = assign_colors_to_bars(coefficient_values, reverse=True) + bar_colors = assign_colors_to_bars(coefficient_values, reverse=reverse_values) bar_colors = list(np.array(bar_colors)[index_sorted]) plt.barh(pos, sorted_column_values, align='center', color=bar_colors)