Add missing sbi import

brian-team · Oct 6, 2023 · 27be7fe · 27be7fe
1 parent 30c292f
commit 27be7fe
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/brian2modelfitting/inferencer.py b/brian2modelfitting/inferencer.py
@@ -653,6 +653,7 @@ def init_inference(self, inference_method, density_estimator_model, prior,
         sbi.inference.NeuralInference
             Instantiated inference object.
         """
+        import sbi.inference
         from sbi.utils.get_nn_models import (posterior_nn,
                                              likelihood_nn,
                                              classifier_nn)