Ruff

Author: arm61

kinisi/fitting.py

@@ -15,7 +15,6 @@
 from scipy.linalg import pinvh
 from scipy.optimize import minimize
 from scipy.stats import uniform
-from scipy.stats._distn_infrastructure import rv_frozen
 
 from kinisi.samples import Samples
 
@@ -61,7 +60,9 @@ def __init__(
                     f'Priors must be a list of length {len(self.parameter_names)}, got {len(priors)} instead.'
                 )
             if not all(hasattr(p, 'logpdf') and hasattr(p, 'ppf') for p in priors):
-                raise ValueError("Priors must provide 'logpdf' and 'ppf' methods (e.g., frozen scipy.stats distributions).")
+                raise ValueError(
+                    "Priors must provide 'logpdf' and 'ppf' methods (e.g., frozen scipy.stats distributions)."
+                )
 
         if self.priors is None:
             # Perform initial fit

kinisi/yeh_hummer.py

@@ -41,7 +41,7 @@ class YehHummer(FittingBase):
 
     :param diffusion: sc.DataArray with diffusion coefficients and box_length coordinate
     :param temperature: Temperature (will be extracted from coords if not provided)
-    :param priors: Optional priors for [D_0, viscosity] parameters using scipy.stats objects 
+    :param priors: Optional priors for [D_0, viscosity] parameters using scipy.stats objects
         (viscosity in Pa*s)
     """