diff --git a/ptable_trends.py b/ptable_trends.py index a1d70eb..563bf01 100644 --- a/ptable_trends.py +++ b/ptable_trends.py @@ -189,9 +189,10 @@ def ptable_plotter( else: color_list[element_index] = to_hex(color_scale[i]) - for k, v in elements["symbol"].iteritems(): - if v in special_elements: - color_list[k] = special_color + if special_elements: + for k, v in elements["symbol"].iteritems(): + if v in special_elements: + color_list[k] = special_color # Define figure properties for visualizing data source = ColumnDataSource(