Fix n_cmaps/cmap check

tupui · tupui · commit 5317f2753b21 · 2023-10-16T18:21:36.000+02:00
diff --git a/src/simdec/visualization.py b/src/simdec/visualization.py
@@ -94,7 +94,7 @@ def palette(
     if cmaps is None:
         cmaps = [mpl.colormaps[cmap] for cmap in SEQUENTIAL_PALETTES[:n_cmaps]]
     else:
-        if len(cmaps) != cmaps:
+        if len(cmaps) != n_cmaps:
             raise ValueError(
                 "Must have the same number of cmaps as the number of first states"
             )

Original file line number	Diff line number	Diff line change
`@@ -94,7 +94,7 @@ def palette(`
`94`	`94`	`if cmaps is None:`
`95`	`95`	`cmaps = [mpl.colormaps[cmap] for cmap in SEQUENTIAL_PALETTES[:n_cmaps]]`
`96`	`96`	`else:`
`97`		`- if len(cmaps) != cmaps:`
	`97`	`+ if len(cmaps) != n_cmaps:`
`98`	`98`	`raise ValueError(`
`99`	`99`	`"Must have the same number of cmaps as the number of first states"`
`100`	`100`	`)`