Move warning filter to code from test

zaxtax · zaxtax · commit 18a67778b8de · 2025-04-12T16:15:48.000+02:00
diff --git a/pymc_extras/inference/pathfinder/pathfinder.py b/pymc_extras/inference/pathfinder/pathfinder.py
@@ -485,6 +485,7 @@ def bfgs_sample_dense(
     shapes: L=batch_size, N=num_params, J=history_size, M=num_samples
     """
 
+    _warnings.simplefilter("ignore", category=FutureWarning)
     N = x.shape[-1]
     IdN = pt.eye(N)[None, ...]
 
@@ -560,6 +561,7 @@ def bfgs_sample_sparse(
     shapes: L=batch_size, N=num_params, J=history_size, M=num_samples
     """
 
+    _warnings.simplefilter("ignore", category=FutureWarning)
     # qr_input: (L, N, 2J)
     qr_input = inv_sqrt_alpha_diag @ beta
     (Q, R), _ = pytensor.scan(fn=pt.nlinalg.qr, sequences=[qr_input], allow_gc=False)
diff --git a/tests/test_pathfinder.py b/tests/test_pathfinder.py
@@ -20,7 +20,6 @@
 
 pytestmark = pytest.mark.filterwarnings(
     "ignore:compile_pymc was renamed to compile:FutureWarning",
-    "ignore:batched_dot is deprecated:FutureWarning",
 )
 
 import pymc_extras as pmx

Original file line number	Diff line number	Diff line change
`@@ -20,7 +20,6 @@`
`20`	`20`
`21`	`21`	`pytestmark = pytest.mark.filterwarnings(`
`22`	`22`	`"ignore:compile_pymc was renamed to compile:FutureWarning",`
`23`		`- "ignore:batched_dot is deprecated:FutureWarning",`
`24`	`23`	`)`
`25`	`24`
`26`	`25`	`import pymc_extras as pmx`