From 1578285a1fd99444a001fce5ceef16c6d13d9f2d Mon Sep 17 00:00:00 2001 From: Ismael Mendoza Date: Sun, 14 Nov 2021 15:28:45 -0500 Subject: [PATCH] should be a public method --- btk/main.py | 4 ++-- btk/survey.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/btk/main.py b/btk/main.py index 337453196..da841442d 100644 --- 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 index 2682ce941..a6e4e07ca 100644 --- 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