use getattr directly

pola-rs · Sep 22, 2024 · e8b5ffd · e8b5ffd
1 parent f792361
commit e8b5ffd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py-polars/polars/dataframe/plotting.py b/py-polars/polars/dataframe/plotting.py
@@ -31,9 +31,9 @@
 
 
 def _maybe_extract_shorthand(encoding: Encoding) -> Encoding:
-    if isinstance(encoding, alt.SchemaBase) and hasattr(encoding, "shorthand"):
+    if isinstance(encoding, alt.SchemaBase):
         # e.g. for `alt.X('x:Q', axis=alt.Axis(labelAngle=30))`, return `'x:Q'`
-        return encoding.shorthand
+        return getattr(encoding, "shorthand", encoding)
     return encoding