make pandera.typing.Series[TYPE] error in polars DataFrameModel more …

…readable (#1588) Signed-off-by: cosmicBboy <niels.bantilan@gmail.com>
unionai-oss · Apr 19, 2024 · 249cab2 · 249cab2
1 parent 8a9d733
commit 249cab2
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/pandera/api/polars/model.py b/pandera/api/polars/model.py
@@ -89,9 +89,19 @@ def _build_columns(  # pylint:disable=too-many-locals
                 columns[field_name] = Column(**column_kwargs)
 
             else:
+                origin_name = (
+                    f"{annotation.origin.__module__}."
+                    f"{annotation.origin.__name__}"
+                )
+                msg = (
+                    " Series[TYPE] annotations are not supported for polars. "
+                    "Use the bare TYPE directly"
+                    if origin_name == "pandera.typing.pandas.Series"
+                    else ""
+                )
                 raise SchemaInitError(
                     f"Invalid annotation '{field_name}: "
-                    f"{annotation.raw_annotation}'"
+                    f"{annotation.raw_annotation}'.{msg}"
                 )
 
         return columns