catch warning in example

pola-rs · Nov 18, 2024 · 5bde56e · 5bde56e
1 parent 1b8ae1b
commit 5bde56e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/docs/source/src/python/user-guide/expressions/categoricals.py b/docs/source/src/python/user-guide/expressions/categoricals.py
@@ -115,6 +115,10 @@
 # --8<-- [end:stringcache-categorical-comparison-physical]
 
 # --8<-- [start:concatenating-categoricals]
+import warnings
+
+from polars.exceptions import CategoricalRemappingWarning
+
 male_bears = pl.DataFrame(
     {
         "species": ["Polar", "Brown", "Panda"],
@@ -130,7 +134,10 @@
     schema_overrides={"species": pl.Categorical},
 )
 
-bears = pl.concat([male_bears, female_bears], how="vertical")
+with warnings.catch_warnings():
+    warnings.filterwarnings("ignore", category=CategoricalRemappingWarning)
+    bears = pl.concat([male_bears, female_bears], how="vertical")
+
 print(bears)
 # --8<-- [end:concatenating-categoricals]