ruff

Author: manuelgloeckler

sbi/inference/potentials/score_based_potential.py

@@ -162,7 +162,7 @@ def gradient(
                 raise NotImplementedError(
                     "Score accumulation for IID data is not yet implemented."
                 )
-               
+
         return score
 
     def get_continuous_normalizing_flow(
@@ -229,4 +229,3 @@ def f(t, x):
         exact=exact,
     )
     return transform
-

sbi/samplers/score/predictors.py

@@ -120,4 +120,3 @@ def predict(self, theta: Tensor, t1: Tensor, t0: Tensor):
         f_backward = f - (1 + self.eta**2) / 2 * g**2 * score
         g_backward = self.eta * g
         return theta - f_backward * dt + g_backward * torch.randn_like(theta) * dt_sqrt
-

tests/score_samplers_test.py

@@ -61,7 +61,6 @@ def _build_gaussian_score_estimator(
     # Note the precondition predicts a correct Gaussian score by default if the neural
     # net predicts 0!
     class DummyNet(torch.nn.Module):
-
         def __init__(self):
             super().__init__()
             self.dummy_param_for_device_detection = torch.nn.Linear(1, 1)