fix imports.

sbi-dev · Feb 16, 2024 · b96a8d9 · b96a8d9
1 parent c84723e
commit b96a8d9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sbi/inference/snle/snle_base.py b/sbi/inference/snle/snle_base.py
@@ -11,7 +11,6 @@
 from torch import Tensor, nn, optim
 from torch.distributions import Distribution
 from torch.nn.utils.clip_grad import clip_grad_norm_
-from torch.utils import data
 from torch.utils.tensorboard.writer import SummaryWriter
 
 from sbi import utils as utils

diff --git a/sbi/inference/snpe/snpe_base.py b/sbi/inference/snpe/snpe_base.py
@@ -26,14 +26,15 @@
     RestrictedPrior,
     check_estimator_arg,
     handle_invalid_x,
+    mask_sims_from_prior,
     nle_nre_apt_msg_on_invalid_x,
     npe_msg_on_invalid_x,
     test_posterior_net_for_multi_d_x,
     validate_theta_and_x,
     warn_if_zscoring_changes_data,
     x_shape_from_simulation,
 )
-from sbi.utils.sbiutils import ImproperEmpirical, mask_sims_from_prior
+from sbi.utils.sbiutils import ImproperEmpirical
 
 
 class PosteriorEstimator(NeuralInference, ABC):