Fix get_style

ISI-MIP · Dec 6, 2023 · f581cdd · f581cdd
1 parent 140bb21
commit f581cdd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/isimip_qa/mixins/plots.py b/isimip_qa/mixins/plots.py
@@ -236,7 +236,11 @@ def get_label(self, i):
 
     def get_style(self, i):
         if self.dimensions:
-            label = self.permutations[i][self.grid:]
+            if self.figs > 0:
+                label = self.permutations[i][self.grid:-self.figs]
+            else:
+                label = self.permutations[i][self.grid:]
+
             if label not in self.styles:
                 index = len(self.styles.keys())
                 color_index = index % len(self.colors)