Skip to content

Commit 9535699

Browse files
authored
AC: improve logging initialization (#2997)
1 parent 8e2aaa9 commit 9535699

File tree

4 files changed

+16
-5
lines changed

4 files changed

+16
-5
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/annotation_converters/convert.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
)
3939
from ..data_analyzer import BaseDataAnalyzer
4040
from .format_converter import BaseFormatConverter
41-
from ..logging import exception
41+
from ..logging import exception, init_logging
4242

4343
DatasetConversionInfo = namedtuple('DatasetConversionInfo',
4444
[
@@ -255,6 +255,7 @@ def make_subset_kaldi(annotation, size, shuffle=True, **kwargs):
255255

256256

257257
def main():
258+
init_logging()
258259
main_argparser = build_argparser()
259260
tm = start_telemetry()
260261
send_telemetry_event(tm, 'annotation_conversion_status', 'started')

tools/accuracy_checker/openvino/tools/accuracy_checker/logging.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -79,9 +79,15 @@ def emit(self, record):
7979
}
8080
}
8181

82-
logging.config.dictConfig(_LOGGING_CONFIGURATION)
83-
8482
_default_logger = logging.getLogger(_DEFAULT_LOGGER_NAME)
83+
_logging_initialized = False
84+
85+
86+
def init_logging():
87+
global _logging_initialized # pylint: disable=W0603
88+
if not _logging_initialized:
89+
logging.config.dictConfig(_LOGGING_CONFIGURATION)
90+
_logging_initialized = True
8591

8692

8793
def _warning_handler(message, category, filename, line_number, *args, **kwargs):
@@ -134,4 +140,5 @@ def add_file_handler(file_name):
134140
}
135141
_LOGGING_CONFIGURATION['handlers']['file_info'] = file_info_handler_config
136142
_LOGGING_CONFIGURATION['loggers'][_DEFAULT_LOGGER_NAME]['handlers'].append('file_info')
137-
logging.config.dictConfig(_LOGGING_CONFIGURATION)
143+
if _logging_initialized:
144+
logging.config.dictConfig(_LOGGING_CONFIGURATION)

tools/accuracy_checker/openvino/tools/accuracy_checker/main.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222

2323
from .argparser import build_arguments_parser
2424
from .config import ConfigReader
25-
from .logging import print_info, add_file_handler, exception
25+
from .logging import print_info, add_file_handler, exception, init_logging
2626
from .evaluators import ModelEvaluator, ModuleEvaluator
2727
from .progress_reporters import ProgressReporter
2828
from .presenters import write_csv_result
@@ -40,6 +40,7 @@
4040

4141

4242
def main():
43+
init_logging()
4344
return_code = 0
4445
args = build_arguments_parser().parse_args()
4546
tm = start_telemetry()

tools/accuracy_checker/openvino/tools/accuracy_checker/misc.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
from pathlib import Path
1919
from .config import ConfigReader
2020
from .evaluators import ModelEvaluator, ModuleEvaluator
21+
from .logging import init_logging
2122

2223
EVALUATION_MODE = {
2324
'models': ModelEvaluator,
@@ -26,6 +27,7 @@
2627

2728

2829
def get_metric_references(config_path, definitions_path=None, additional_info=None, return_header=True):
30+
init_logging()
2931
args = {'config': Path(config_path), 'definitions': Path(definitions_path) if definitions_path else None}
3032
if additional_info:
3133
args.update(additional_info)

0 commit comments

Comments
 (0)