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 0000000..f5e4cb4 Binary files /dev/null and b/test_autoconf/files/array_out.fits differ