adapt signature of Adapter.standardize

Author: vpratz

bayesflow/adapters/adapter.py

@@ -813,6 +813,8 @@ def standardize(
         self,
         include: str | Sequence[str] = None,
         *,
+        mean: int | float | np.ndarray,
+        std: int | float | np.ndarray,
         predicate: Predicate = None,
         exclude: str | Sequence[str] = None,
         **kwargs,
@@ -821,10 +823,14 @@ def standardize(
 
         Parameters
         ----------
-        predicate : Predicate, optional
-            Function that indicates which variables should be transformed.
         include : str or Sequence of str, optional
             Names of variables to include in the transform.
+        mean : int or float
+            Specifies the mean (location) of the transform.
+        std : int or float
+            Specifies the standard deviation (scale) of the transform.
+        predicate : Predicate, optional
+            Function that indicates which variables should be transformed.
         exclude : str or Sequence of str, optional
             Names of variables to exclude from the transform.
         **kwargs :

bayesflow/adapters/transforms/standardize.py

@@ -25,8 +25,7 @@ class Standardize(ElementwiseTransform):
 
     Examples
     --------
-    >>> adapter = (
-            bf.adapters.Adapter().standardize(include="beta", mean=5, std=10)
+    >>> adapter = bf.Adapter().standardize(include="beta", mean=5, std=10)
     """
 
     def __init__(