fix lints :)

dirmeier · Feb 29, 2024 · 29a2a9b · 29a2a9b
1 parent 16b9b3b
commit 29a2a9b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sbijax/_src/scmpe.py b/sbijax/_src/scmpe.py
@@ -34,7 +34,7 @@ def _discretization_schedule(n_iter, max_iter=1000):
  return nk
 
 
-# pylint: disable=too-many-locals
+# pylint: disable=too-many-locals,too-many-arguments
 def _consistency_loss(
  params,
  ema_params,
@@ -223,6 +223,7 @@ def _init_params(self, rng_key, **init_data):
  )
  return params
 
+ # pylint: disable=arguments-differ
  def _validation_loss(self, rng_key, params, ema_params, n_iter, val_iter):
  loss_fn = jax.jit(
  partial(