Update.

diff --git a/grids.py b/grids.py
index 9462f87..b40d532 100755 (executable)
--- a/grids.py
+++ b/grids.py
@@ -23,9 +23,9 @@ class Grids(problem.Problem):
         ("red", [255, 0, 0]),
         ("green", [0, 192, 0]),
         ("blue", [0, 0, 255]),
-        ("orange", [255, 192, 0]),
+        ("yellow", [255, 224, 0]),
         ("cyan", [0, 255, 255]),
-        ("violet", [255, 0, 255]),
+        ("violet", [224, 128, 255]),
         ("lightgreen", [192, 255, 192]),
         ("brown", [165, 42, 42]),
         ("lightblue", [192, 192, 255]),
@@ -34,7 +34,6 @@ class Grids(problem.Problem):
 
     def __init__(self, device=torch.device("cpu")):
         self.colors = torch.tensor([c for _, c in self.named_colors])
-        self.name2color = dict([(p[0], i) for i, p in enumerate(self.named_colors)])
         self.height = 10
         self.width = 10
         self.device = device