Merge pull request #38 from dirac-institute/issue/36/kbmod-stderr-std…

…out-logging
dirac-institute · Aug 28, 2024 · 1a834a5 · 1a834a5
2 parents c4cb676 + 292f219
commit 1a834a5
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 0 deletions.
diff --git a/src/kbmod_wf/task_impls/ic_to_wu.py b/src/kbmod_wf/task_impls/ic_to_wu.py
@@ -48,6 +48,7 @@ def __init__(
         self.wu_filepath = wu_filepath
         self.runtime_config = runtime_config
         self.logger = logger
+        kbmod._logging.basicConfig(level=self.logger.level)
 
         self.overwrite = self.runtime_config.get("overwrite", False)
         self.search_config_filepath = self.runtime_config.get("search_config_filepath", None)

diff --git a/src/kbmod_wf/task_impls/kbmod_search.py b/src/kbmod_wf/task_impls/kbmod_search.py
@@ -49,6 +49,7 @@ def __init__(
         self.runtime_config = runtime_config
         self.result_filepath = result_filepath
         self.logger = logger
+        kbmod._logging.basicConfig(level=self.logger.level)
 
         self.search_config_filepath = self.runtime_config.get("search_config_filepath", None)
         self.results_directory = os.path.dirname(self.result_filepath)

diff --git a/src/kbmod_wf/task_impls/reproject_multi_chip_multi_night_wu.py b/src/kbmod_wf/task_impls/reproject_multi_chip_multi_night_wu.py
@@ -1,3 +1,4 @@
+import kbmod
 from kbmod.work_unit import WorkUnit
 from kbmod.reprojection_utils import transform_wcses_to_ebd
 
@@ -67,6 +68,7 @@ def __init__(
         self.reprojected_wu_filepath = reprojected_wu_filepath
         self.runtime_config = runtime_config
         self.logger = logger
+        kbmod._logging.basicConfig(level=self.logger.level)
 
         self.overwrite = self.runtime_config.get("overwrite", False)
         self.search_config = self.runtime_config.get("search_config", None)

diff --git a/src/kbmod_wf/task_impls/reproject_single_chip_single_night_wu.py b/src/kbmod_wf/task_impls/reproject_single_chip_single_night_wu.py
@@ -1,3 +1,4 @@
+import kbmod
 from kbmod.work_unit import WorkUnit
 
 import kbmod.reprojection as reprojection
@@ -56,6 +57,7 @@ def __init__(
         self.reprojected_wu_filepath = reprojected_wu_filepath
         self.runtime_config = runtime_config
         self.logger = logger
+        kbmod._logging.basicConfig(level=self.logger.level)
 
         # Default to 8 workers if not in the config. Value must be 0<num workers<65.
         self.n_workers = max(1, min(self.runtime_config.get("n_workers", 8), 64))