Merge branch 'main' into feature/truncated_gaussian

Author: Jammy2211

autofit/non_linear/search/abstract_search.py

@@ -680,6 +680,7 @@ def start_resume_fit(self, analysis: Analysis, model: AbstractPriorModel) -> Res
             model=model,
             analysis=analysis,
         )
+
         samples = self.perform_update(
             model=model,
             analysis=analysis,

autofit/non_linear/search/nest/dynesty/search/static.py

@@ -6,6 +6,8 @@
 from dynesty import NestedSampler as StaticSampler
 
 from autofit.database.sqlalchemy_ import sa
+from autofit import jax_wrapper
+
 from autofit.mapper.prior_model.abstract import AbstractPriorModel
 
 from .abstract import AbstractDynesty, prior_transform

autofit/non_linear/search/nest/nautilus/search.py

@@ -338,7 +338,7 @@ def call_search(self, search_internal, model, analysis, fitness):
                     search_internal=search_internal
                 )
 
-        return search_internal, fitness
+        return search_internal
 
     def fit_mpi(self, fitness, model, analysis, checkpoint_exists: bool):
         """
@@ -396,7 +396,7 @@ def fit_mpi(self, fitness, model, analysis, checkpoint_exists: bool):
                 **self.config_dict_run,
             )
 
-        return search_internal, fitness
+        return search_internal
 
     def iterations_from(
         self, search_internal