Better deprecation warning message

pymc-devs · Jan 9, 2025 · ce09972 · ce09972
1 parent 8e9e7b4
commit ce09972
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/pymc/variational/inference.py b/pymc/variational/inference.py
@@ -84,7 +84,7 @@ def run_profiling(self, n=1000, score=None, **kwargs):
         score = self._maybe_score(score)
         if "fn_kwargs" in kwargs:
             warnings.warn(
-                DeprecationWarning, "fn_kwargs is deprecated, please use compile_kwargs instead"
+                "fn_kwargs is deprecated, please use compile_kwargs instead", DeprecationWarning
             )
             compile_kwargs = kwargs.pop("fn_kwargs")
         else:

diff --git a/pymc/variational/opvi.py b/pymc/variational/opvi.py
@@ -379,7 +379,11 @@ def step_function(
         `pytensor.function`
         """
         if fn_kwargs is not None:
-            warnings.warn("`fn_kwargs` is deprecated and will be removed in future versions")
+            warnings.warn(
+                "`fn_kwargs` is deprecated and will be removed in future versions. Use "
+                "`compile_kwargs` instead.",
+                DeprecationWarning,
+            )
             compile_kwargs = fn_kwargs
 
         if compile_kwargs is None:
@@ -428,7 +432,11 @@ def score_function(
         pytensor.function
         """
         if fn_kwargs is not None:
-            warnings.warn("`fn_kwargs` is deprecated and will be removed in future versions")
+            warnings.warn(
+                "`fn_kwargs` is deprecated and will be removed in future versions. Use "
+                "`compile_kwargs` instead",
+                DeprecationWarning,
+            )
             compile_kwargs = fn_kwargs
 
         if compile_kwargs is None: