dont override global

apache · Nov 20, 2024 · 769d9cf · 769d9cf
1 parent 12e87a4
commit 769d9cf
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/pyiceberg/utils/deprecated.py b/pyiceberg/utils/deprecated.py
@@ -55,11 +55,10 @@ def deprecation_message(deprecated_in: str, removed_in: str, help_message: Optio
 
 
 def _deprecation_warning(message: str) -> None:
-    warnings.simplefilter("always", DeprecationWarning)  # turn off filter
-
-    warnings.warn(
-        message,
-        category=DeprecationWarning,
-        stacklevel=2,
-    )
-    warnings.simplefilter("default", DeprecationWarning)  # reset filter
+    with warnings.catch_warnings():  # temporarily override warning handling
+        warnings.simplefilter("always", DeprecationWarning)  # turn off filter
+        warnings.warn(
+            message,
+            category=DeprecationWarning,
+            stacklevel=2,
+        )