X-Git-Url: https://git.arvados.org/lightning.git/blobdiff_plain/219c3aa996193dacd8e38f7aa4e3d0d32b84f403..ba5dea916505730ed44af0655c820bd3d767be1e:/plot.go

diff --git a/plot.go b/plot.go
index 4941f6fe8b..5e2eb63845 100644
--- a/plot.go
+++ b/plot.go
@@ -116,7 +116,10 @@ if sys.argv[2]:
         'PJL': 'blueviolet',
     }
     for fnm in sorted(labels.keys()):
-        colors.append(labelcolors[labels[fnm]])
+        if labels[fnm] in labelcolors:
+            colors.append(labelcolors[labels[fnm]])
+        else:
+            colors.append('black')
 
 from matplotlib.figure import Figure
 from matplotlib.patches import Polygon