ruff tests

sbi-dev · Apr 9, 2024 · 8b00209 · 8b00209
1 parent f9a90d0
commit 8b00209
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/density_estimator_test.py b/tests/density_estimator_test.py
@@ -10,13 +10,11 @@
 from torch import eye, zeros
 from torch.distributions import MultivariateNormal
 
+from sbi.neural_nets import build_mnle
+from sbi.neural_nets.categorial import build_categoricalmassestimator
 from sbi.neural_nets.density_estimators.shape_handling import (
     reshape_to_sample_batch_event,
 )
-from sbi.neural_nets.flow import build_zuko_maf, build_maf
-from sbi.neural_nets.mdn import build_mdn
-from sbi.neural_nets import build_mnle
-from sbi.neural_nets.categorial import build_categoricalmassestimator
 from sbi.neural_nets.embedding_nets import CNNEmbedding
 from sbi.neural_nets.flow import (
     build_maf,
@@ -32,6 +30,7 @@
     build_zuko_sospf,
     build_zuko_unaf,
 )
+from sbi.neural_nets.mdn import build_mdn
 
 
 def get_batch_input(nsamples: int, input_dims: int) -> torch.Tensor: