Fixes

vandeplaslab · Feb 17, 2025 · a196fec · a196fec
1 parent 70745dd
commit a196fec
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/koyo/click.py b/src/koyo/click.py
@@ -465,6 +465,8 @@ def parse_extra_args(extra_args: tuple[str, ...] | None) -> dict[str, ty.Any]:
             continue
         name, value = parse_arg(arg, "")
         if name in kwargs:
+            if name in kwargs and not isinstance(kwargs[name], list):
+                kwargs[name] = [kwargs[name]]
             if isinstance(kwargs[name], list):
                 kwargs[name].append(value)
             elif isinstance(kwargs[name], (str, int, float, bool)):

diff --git a/src/koyo/visuals.py b/src/koyo/visuals.py
@@ -646,6 +646,15 @@ def fix_style(style: str) -> str:
     return style
 
 
+def shorten_style(style: str) -> str:
+    """Shorten style name."""
+    style = style.replace("seaborn-v0_8-", "s-")  # seaborn style is too long
+    style = style.replace("seaborn-", "s-")
+    style = style.replace("seaborn", "s")
+    style = style.replace("dark_background", "dark")
+    return style
+
+
 def _override_seaborn_heatmap_annotations():
     from seaborn.matrix import _HeatMapper