Update polars_engine.py

unionai-oss · Sep 21, 2024 · c3a3715 · c3a3715
1 parent 6e8a657
commit c3a3715
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pandera/engines/polars_engine.py b/pandera/engines/polars_engine.py
@@ -696,7 +696,7 @@ class Enum(DataType):
 
     def __init__(  # pylint:disable=super-init-not-called
         self,
-        categories: pl.Series | Iterable[str] | None = None,
+        categories: Union[pl.Series, Iterable[str], None] = None,
     ) -> None:
         object.__setattr__(self, "categories", categories)
         object.__setattr__(self, "type", pl.Enum(categories=categories))