From 8a55bf13e5cfa11f776916371b9f2e6144b28d20 Mon Sep 17 00:00:00 2001 From: Jammy2211 Date: Sun, 12 Apr 2026 13:11:09 +0100 Subject: [PATCH] refactor: separate PYAUTOFIT_TEST_MODE into distinct PYAUTO_* env vars Split the catch-all PYAUTOFIT_TEST_MODE into purpose-specific variables: - PYAUTO_TEST_MODE (sampler speedup, levels 0-3) - PYAUTO_SKIP_FIT_OUTPUT (pre/post-fit I/O, VRAM, result text) - PYAUTO_SKIP_VISUALIZATION (fit visualization and plotting) - PYAUTO_SKIP_CHECKS (mesh validation, position resampling, weight thresholds) Also renames: - PYAUTOARRAY_OUTPUT_MODE -> PYAUTO_OUTPUT_MODE - PYAUTO_WORKSPACE_SMALL_DATASETS -> PYAUTO_SMALL_DATASETS Co-Authored-By: Claude Opus 4.6 --- autoconf/__init__.py | 2 +- autoconf/test_mode.py | 32 ++++++++++++++++++++++++++++- test_autoconf/files/array_out.fits | Bin 0 -> 5760 bytes 3 files changed, 32 insertions(+), 2 deletions(-) create mode 100644 test_autoconf/files/array_out.fits diff --git a/autoconf/__init__.py b/autoconf/__init__.py index e60f224..ead3933 100644 --- a/autoconf/__init__.py +++ b/autoconf/__init__.py @@ -10,7 +10,7 @@ from .setup_colab import for_autolens from .setup_notebook import setup_notebook -from .test_mode import test_mode_level, is_test_mode +from .test_mode import test_mode_level, is_test_mode, skip_fit_output, skip_visualization, skip_checks __version__ = "2026.4.5.3" diff --git a/autoconf/test_mode.py b/autoconf/test_mode.py index 537d05a..40d7ddb 100644 --- a/autoconf/test_mode.py +++ b/autoconf/test_mode.py @@ -10,7 +10,7 @@ def test_mode_level(): 2 = bypass sampler entirely, call likelihood once 3 = bypass sampler entirely, skip likelihood call """ - return int(os.environ.get("PYAUTOFIT_TEST_MODE", "0")) + return int(os.environ.get("PYAUTO_TEST_MODE", "0")) def is_test_mode(): @@ -18,3 +18,33 @@ def is_test_mode(): Return True if any test mode is active. """ return test_mode_level() > 0 + + +def skip_fit_output(): + """ + Return True if fit I/O should be skipped. + + Controls: pre/post-fit output, VRAM profiling, result info text, + likelihood function checks. + """ + return os.environ.get("PYAUTO_SKIP_FIT_OUTPUT", "0") == "1" + + +def skip_visualization(): + """ + Return True if fit visualization should be skipped. + + Controls: Visualizer.should_visualize, plot decorators, + quantity visualizers. + """ + return os.environ.get("PYAUTO_SKIP_VISUALIZATION", "0") == "1" + + +def skip_checks(): + """ + Return True if validation checks should be skipped. + + Controls: mesh pixel validation (hilbert), position resampling, + inversion position exceptions, sample weight thresholds. + """ + return os.environ.get("PYAUTO_SKIP_CHECKS", "0") == "1" diff --git a/test_autoconf/files/array_out.fits b/test_autoconf/files/array_out.fits new file mode 100644 index 0000000000000000000000000000000000000000..f5e4cb4bc8d10eeb8b2afb04a86c5894d8fe9e7e GIT binary patch literal 5760 zcmeH@K?{N~7>4^P-y!f6kxIA1LW4ntppXz9RvL70i)n}ceZv-dn3U(<@343K`1Y_p z3ERcnh-g8UN@+?LDXvnL71fe9EKR7YxyZT7E%UmyW@*eaD{nIQHIFOBpOo{OQ~kUb zc1qjbuQHbCqxF5R=7~pnE)>b@hkPpwQQf4dT1aQt@Mb+cuQT}Lw&A%wJl!8!D|5m6 z-XZsbFa`G