update colors

data.py

@@ -53,10 +53,12 @@ def make_colors() -> UInt8[np.ndarray, "n 3"]:
     colors[4] = np.array([151, 204, 4], dtype=np.uint8)
     colors[13] = np.array([104, 139, 88], dtype=np.uint8)
     colors[16] = np.array([54, 48, 32], dtype=np.uint8)
+    colors[21] = np.array([120, 202, 210], dtype=np.uint8)  # water
     colors[26] = np.array([45, 125, 210], dtype=np.uint8)
     colors[29] = np.array([116, 142, 84], dtype=np.uint8)
     colors[46] = np.array([238, 185, 2], dtype=np.uint8)
     colors[52] = np.array([88, 91, 86], dtype=np.uint8)
+    colors[60] = np.array([72, 99, 156], dtype=np.uint8)  # river
     colors[72] = np.array([76, 46, 5], dtype=np.uint8)
     colors[94] = np.array([12, 15, 10], dtype=np.uint8)