Fix expanded states

Author: tupui

panel/simdec_app.py

@@ -188,7 +188,7 @@ def palette_(states: list[list[str]], colors_picked: list[list[float]]):
     cmaps = [single_color_to_colormap(color_picked) for color_picked in colors_picked]
     # Reverse order as in figures high values take the first colors
     states = [len(states_) for states_ in states]
-    return sd.palette(states[::-1], cmaps=cmaps[::-1])
+    return sd.palette(states, cmaps=cmaps[::-1])
 
 
 @pn.cache

src/simdec/decomposition.py

@@ -22,6 +22,8 @@ def states_expansion(states: list[int], inputs: pd.DataFrame) -> list[list[str]]
                 expanded_states.append(["low", "high"])
             elif state == 3:
                 expanded_states.append(["low", "medium", "high"])
+            else:
+                expanded_states.append([i for i in range(state)])
         else:
             expanded_states.append(state)