WIP: Add __all__ for non-test modules

Author: pllim

jwst/__init__.py

@@ -4,6 +4,8 @@
 
 from importlib.metadata import version
 
+__all__ = ["__version__", "__version_commit__"]
+
 __version__ = version(__name__)
 
 _regex_git_hash = re.compile(r".*\+g(\w+)")

jwst/ami/ami_analyze.py

@@ -12,6 +12,8 @@
 log = logging.getLogger(__name__)
 log.setLevel(logging.DEBUG)
 
+__all__ = ["apply_lg_plus"]
+
 
 def apply_lg_plus(
     input_model,

jwst/ami/ami_average.py

@@ -8,6 +8,8 @@
 log = logging.getLogger(__name__)
 log.setLevel(logging.DEBUG)
 
+__all__ = ["average_lg"]
+
 
 def average_lg(lg_products):
     """

jwst/ami/ami_normalize.py

@@ -9,6 +9,8 @@
 log = logging.getLogger(__name__)
 log.addHandler(logging.NullHandler())
 
+__all__ = ["normalize_lg"]
+
 
 def normalize_lg(target_model, reference_model):
     """

jwst/ami/bp_fix.py

@@ -42,6 +42,16 @@
 DO_NOT_USE = dqflags.pixel["DO_NOT_USE"]
 JUMP_DET = dqflags.pixel["JUMP_DET"]
 
+__all__ = [
+    "create_wavelengths",
+    "calc_pupil_support",
+    "transform_image",
+    "calcpsf",
+    "bad_pixels",
+    "fourier_corr",
+    "fix_bad_pixels",
+]
+
 
 def create_wavelengths(filtername):
     """