AMD

eelbrain/plot/_styles.py

@@ -164,10 +164,10 @@ def find_cell_styles(
             out = {None: colors}
     elif isinstance(colors, (list, tuple)):
         if len(colors) < len(cells):
-            raise ValueError(f"colors={colors!r}: only {len(colors)} colors for {len(cells)} cells.")
+            raise ValueError(f"{colors=}: only {len(colors)} colors for {len(cells)} cells.")
         out = dict(zip(cells, colors))
     elif isinstance(colors, dict):
-        out = colors.copy()
+        out = to_styles_dict(colors)
         missing = [cell for cell in cells if cell not in out]
         if missing:
             if fallback:
@@ -182,6 +182,7 @@ def find_cell_styles(
                             break
             if missing:
                 raise KeysMissing(missing, 'colors', colors)
+        return out
     elif colors is None or isinstance(colors, str):
         if default_cells is not None:
             default_colors = find_cell_styles(default_cells, colors)
@@ -469,4 +470,4 @@ def soft_threshold_colormap(
 
 ColorArg = Union[str, Sequence[float]]
 ColorsArg = Union[ColorArg, Dict[CellArg, ColorArg], Sequence[ColorArg]]
-StylesDict = Dict[CellArg, Style]
+StylesDict = Dict[Optional[CellArg], Style]

eelbrain/plot/tests/test_uts.py

@@ -24,7 +24,7 @@ def test_uts_stat():
     assert p.figure.axes[1].get_title() == 'b1'
     assert set(p._LegendMixin__handles) == {'a0', 'a1'}
     p.close()
-    p = plot.UTSStat('uts', 'A', 'B', match='rm', ds=ds, pool_error=False)
+    p = plot.UTSStat('uts', 'A', 'B', match='rm', ds=ds, pool_error=False, colors={'a0': 'red', 'a1': 'k'})
     assert [len(pl.stat_plots) for pl in p._plots] == [2, 2]
     p.close()