Better handling for preliz warnings

Author: jessegrabowski

pymc_extras/prior.py

@@ -1190,8 +1190,6 @@ def constrain(self, lower: float, upper: float, mass: float = 0.95, kwargs=None)
             ).constrain(lower=0.5, upper=0.8)
 
         """
-        import warnings
-
         from preliz import maxent
 
         if self.transform:
@@ -1201,12 +1199,10 @@ def constrain(self, lower: float, upper: float, mass: float = 0.95, kwargs=None)
             kwargs = {}
             kwargs.setdefault("plot", False)
 
-        with warnings.catch_warnings():
-            warnings.simplefilter("ignore")
-            if kwargs["plot"]:
-                new_parameters = maxent(self.preliz, lower, upper, mass, **kwargs)[0].params_dict
-            else:
-                new_parameters = maxent(self.preliz, lower, upper, mass, **kwargs).params_dict
+        if kwargs["plot"]:
+            new_parameters = maxent(self.preliz, lower, upper, mass, **kwargs)[0].params_dict
+        else:
+            new_parameters = maxent(self.preliz, lower, upper, mass, **kwargs).params_dict
 
         return Prior(
             self.distribution,

pyproject.toml

@@ -104,7 +104,10 @@ filterwarnings =[
     'ignore:os\.fork\(\) was called\.:RuntimeWarning',
 
     # Preliz needs to update for pytensor > 2.35
-    'ignore:.*`pytensor\.graph\.basic\.ancestors`.*`pytensor\.graph\.traversal\.ancestors`.*:FutureWarning:^preliz(\.|$)',
+    'ignore:.*`pytensor\.graph\.basic\.ancestors`.*`pytensor\.graph\.traversal\.ancestors`.*:FutureWarning:^preliz(\.$)',
+
+    # Preliz triggers pytensor broadcastable deprecation internally
+    'ignore:The `broadcastable` keyword is deprecated:DeprecationWarning:pytensor',
 
     # OpenMP library warning on windows CI
     'ignore::RuntimeWarning:threadpoolctl',