Update ptable_trends.py

Andrew-S-Rosen · Feb 14, 2022 · a784182 · a784182
1 parent c933684
commit a784182
Showing 1 changed file with 4 additions and 3 deletions.
diff --git 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(