Bump import memory.

[lightning.git] / plot.go

diff --git a/plot.go b/plot.go
index 4941f6fe8b6d9e1bf7ac0db795af7da766de6804..73f1f3c3d446dd653b2358f082cb167f2022f88d 100644 (file)
--- a/plot.go
+++ b/plot.go
@@ -37,7 +37,7 @@ func (cmd *pythonPlot) RunCommand(prog string, args []string, stdin io.Reader, s
                Name:        "lightning plot",
                Client:      arvados.NewClientFromEnv(),
                ProjectUUID: *projectUUID,
-               RAM:         1 << 30,
+               RAM:         4 << 30,
                VCPUs:       1,
                Priority:    *priority,
                Mounts: map[string]map[string]interface{}{
@@ -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