Move logging config into client code

spring-media · Feb 27, 2023 · c829fba · c829fba
1 parent a06d714
commit c829fba
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 12 deletions.
diff --git a/dp/utils/logging.py b/dp/utils/logging.py
@@ -1,16 +1,4 @@
-import logging.config
-import logging.config
-import os
 from logging import Logger, getLogger
-from pathlib import Path
-
-import dp.configs
-from dp.utils.io import read_config
-
-main_dir = os.path.dirname(os.path.abspath(dp.configs.__file__))
-config_file_path = Path(main_dir) / 'logging.yaml'
-config = read_config(config_file_path)
-logging.config.dictConfig(config)
 
 
 def get_logger(name: str) -> Logger:

diff --git a/dp/configs/logging.yaml → logging.yaml b/dp/configs/logging.yaml → logging.yaml
diff --git a/run_training.py b/run_training.py
@@ -1,8 +1,15 @@
+import logging.config
+from pathlib import Path
 import torch
 import torch.multiprocessing as mp
 
 from dp.preprocess import preprocess
 from dp.train import train
+from dp.utils.io import read_config
+
+config_file_path = Path('logging.yaml')
+config = read_config(config_file_path)
+logging.config.dictConfig(config)
 
 if __name__ == '__main__':