aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/__init__.py
blob: 5a6fdc419ae67d41f15dd049a6a6b6712119a2e9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Init of MLIA."""
import logging
import os
from importlib.metadata import version

# redirect warnings to logging
logging.captureWarnings(True)


# as TensorFlow tries to configure root logger
# it should be configured before importing TensorFlow
root_logger = logging.getLogger()
root_logger.addHandler(logging.NullHandler())


# disable TensorFlow warning messages
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"

__version__ = version("mlia")