should be a public method

LSSTDESC · Nov 14, 2021 · 1578285 · 1578285
1 parent a4cc1a0
commit 1578285
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/btk/main.py b/btk/main.py
@@ -7,12 +7,12 @@
 from btk.measure import available_measure_functions
 from btk.measure import MeasureGenerator
 from btk.metrics import MetricsGenerator
-from btk.survey import _get_survey_from_cfg
+from btk.survey import get_survey_from_cfg
 
 
 def main(cfg: OmegaConf):
     """Run BTK from end-to-end using a hydra configuration object."""
-    surveys = [_get_survey_from_cfg(cfg.surveys[survey_name]) for survey_name in cfg.surveys]
+    surveys = [get_survey_from_cfg(cfg.surveys[survey_name]) for survey_name in cfg.surveys]
 
     # get draw blends generator.
     draw_blend_generator = instantiate(cfg.draw_blends, surveys=surveys)

diff --git a/btk/survey.py b/btk/survey.py
@@ -63,7 +63,7 @@
     extinction (float): Exponential extinction coefficient for atmospheric absorption"""
 
 
-def _get_survey_from_cfg(survey_conf: OmegaConf):
+def get_survey_from_cfg(survey_conf: OmegaConf):
     """Creates the corresponding `btk.survey.Survey` object using the information from config file.
 
     Args:
@@ -129,7 +129,7 @@ def get_surveys(names="Rubin", overrides: Iterable = ()):
         cfg = compose("config", overrides=overrides)
     for survey_name in cfg.surveys:
         survey_conf = cfg.surveys[survey_name]
-        surveys.append(_get_survey_from_cfg(survey_conf))
+        surveys.append(get_survey_from_cfg(survey_conf))
     if len(surveys) == 1:
         return surveys[0]
     return surveys