applied keenerd#12 (comment)

eric-foy · Oct 31, 2020 · 2627cb6 · 2627cb6
1 parent f7a6c03
commit 2627cb6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/heatmap/heatmap.py b/heatmap/heatmap.py
@@ -346,6 +346,8 @@ def rgb_fn(palette, min_z, max_z):
     def rgb_inner(z):
         tone = (z - min_z) / (max_z - min_z)
         tone_scaled = int(tone * (len(palette)-1))
+        if tone_scaled >= len(palette):
+            tone_scaled = len(palette)-1
         return palette[tone_scaled]
     return rgb_inner