Update polars_engine.py

unionai-oss · Sep 21, 2024 · 6e8a657 · 6e8a657
1 parent dc4d7f0
commit 6e8a657
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pandera/engines/polars_engine.py b/pandera/engines/polars_engine.py
@@ -719,7 +719,7 @@ def check(
 
         return (
             self.type == pandera_dtype.type
-            and (self.type.categories == getattr(pandera_dtype, "categories", None)).all()
+            and (self.type.categories == pandera_dtype.categories).all()  # type: ignore
         )