diff --git a/autolens/__init__.py b/autolens/__init__.py index 5ffbd1898..6fb6aa1a2 100644 --- a/autolens/__init__.py +++ b/autolens/__init__.py @@ -58,6 +58,7 @@ ) from autogalaxy.gui.clicker import Clicker from autogalaxy.gui.scribbler import Scribbler +from autogalaxy.gui.display_util import mask_2d_regridded_from from autogalaxy.galaxy.galaxy import Galaxy from autogalaxy.galaxy.galaxies import Galaxies from autogalaxy.galaxy.mass_field import MassField diff --git a/test_autolens/test_gui_reexports.py b/test_autolens/test_gui_reexports.py new file mode 100644 index 000000000..be93f88f9 --- /dev/null +++ b/test_autolens/test_gui_reexports.py @@ -0,0 +1,8 @@ +"""Sanity check that autogalaxy GUI helpers are re-exported under al.*.""" +import autolens as al + + +def test__mask_2d_regridded_from_is_exported_at_package_level(): + from autogalaxy.gui import display_util as du + + assert al.mask_2d_regridded_from is du.mask_2d_regridded_from