add valueerror

martinvonk · Oct 25, 2024 · f6fa1de · f6fa1de
1 parent 8d6401a
commit f6fa1de
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/spei/plot.py b/src/spei/plot.py
@@ -168,6 +168,8 @@ def _get_cmap(self) -> mpl.colors.Colormap:
             return self.vik()
         elif "lajolla" in self.name:
             return self.lajolla()
+        else:
+            raise ValueError(f"Invalid colormap name: {self.name}")
 
     @staticmethod
     def cmap_from_list(