fix: Support already parsed shorthand in FieldChannelMixin

Ensures the same output from ```py alt.Y("count()") alt.Y(alt.agg.count()) ```
dangotbanned · Jul 26, 2024 · 5929a5e · 5929a5e
1 parent 766d1bf
commit 5929a5e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/altair/vegalite/v5/schema/channels.py b/altair/vegalite/v5/schema/channels.py
@@ -169,7 +169,7 @@ def to_dict(
 
         if shorthand is Undefined:
             parsed = {}
-        elif isinstance(shorthand, str):
+        elif isinstance(shorthand, (str, dict)):
             parsed = parse_shorthand(shorthand, data=context.get("data", None))
             type_required = "type" in self._kwds  # type: ignore[attr-defined]
             type_in_shorthand = "type" in parsed

diff --git a/tools/generate_schema_wrapper.py b/tools/generate_schema_wrapper.py
@@ -99,7 +99,7 @@ def to_dict(
 
         if shorthand is Undefined:
             parsed = {}
-        elif isinstance(shorthand, str):
+        elif isinstance(shorthand, (str, dict)):
             parsed = parse_shorthand(shorthand, data=context.get("data", None))
             type_required = "type" in self._kwds  # type: ignore[attr-defined]
             type_in_shorthand = "type" in parsed