Merge pull request #979 from rhayes777/feature/test_mode_samples_weig…

…ht_threshold Feature/test mode samples weight threshold
rhayes777 · Mar 22, 2024 · fa923f7 · fa923f7
2 parents 6528a2c + 4a96585
commit fa923f7
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/autofit/non_linear/search/abstract_search.py b/autofit/non_linear/search/abstract_search.py
@@ -926,6 +926,9 @@ def perform_update(
 
             samples_weight_threshold = conf.instance["output"]["samples_weight_threshold"]
 
+            if os.environ.get("PYAUTOFIT_TEST_MODE") == "1":
+                samples_weight_threshold = None
+
             if samples_weight_threshold is not None:
                 samples_for_csv = samples_for_csv.samples_above_weight_threshold_from(
                     weight_threshold=samples_weight_threshold

diff --git a/autofit/non_linear/search/optimize/pyswarms/search/abstract.py b/autofit/non_linear/search/optimize/pyswarms/search/abstract.py
@@ -159,7 +159,7 @@ def _fit(self, model: AbstractPriorModel, analysis):
                 "Resuming PySwarms non-linear search (previous samples found)."
             )
 
-        except (FileNotFoundError, TypeError):
+        except (FileNotFoundError, TypeError, AttributeError):
 
             unit_parameter_lists, parameter_lists, log_posterior_list = self.initializer.samples_from_model(
                 total_points=self.config_dict_search["n_particles"],

diff --git a/docs/api/plot.rst b/docs/api/plot.rst
@@ -22,8 +22,6 @@ Plotters
    :template: custom-class-template.rst
    :recursive:
 
-   DynestyPlotter
-   UltraNestPlotter
-   EmceePlotter
-   ZeusPlotter
-   PySwarmsPlotter
+   NestPlotter
+   MCMCPlotter
+   OptimizePlotter