From 11e18d4b6ccca4b76ea2535d7733d9a845ed61c3 Mon Sep 17 00:00:00 2001 From: Jammy2211 Date: Wed, 8 Apr 2026 13:10:34 +0100 Subject: [PATCH] Re-enable model.info output in test mode 2+ model.info was being skipped entirely for PYAUTOFIT_TEST_MODE >= 2, but benchmarking shows it only takes ~0.18s. Re-enable it so smoke tests can verify the model composition is correct. - Remove early return guard from model.info property - Add model.info file write in the test-mode-2+ else branch Co-Authored-By: Claude Opus 4.6 --- autofit/mapper/prior_model/abstract.py | 5 ----- autofit/non_linear/search/abstract_search.py | 3 +++ 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/autofit/mapper/prior_model/abstract.py b/autofit/mapper/prior_model/abstract.py index 2e1f9f3ce..f36ddbb28 100644 --- a/autofit/mapper/prior_model/abstract.py +++ b/autofit/mapper/prior_model/abstract.py @@ -1778,11 +1778,6 @@ def info(self) -> str: parameter of the overall model. This information is extracted from each priors *model_info* property. """ - from autofit.non_linear.test_mode import test_mode_level - - if test_mode_level() >= 2: - return f"Total Free Parameters = {self.prior_count}\n\n[test mode — info skipped]" - formatter = TextFormatter(line_length=info_whitespace()) for t in find_groups( diff --git a/autofit/non_linear/search/abstract_search.py b/autofit/non_linear/search/abstract_search.py index 8e9f0a43b..afc8ea5e5 100644 --- a/autofit/non_linear/search/abstract_search.py +++ b/autofit/non_linear/search/abstract_search.py @@ -504,6 +504,9 @@ class represented by model M and gives a score for their fitness. model=model, info=info, ) + else: + if hasattr(self.paths, '_save_model_info'): + self.paths._save_model_info(model=model) if not self.paths.is_complete: result = self.start_resume_fit(