aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/__init__.py
diff options
context:
space:
mode:
authorDiego Russo <diego.russo@arm.com>2022-05-30 13:34:14 +0100
committerDiego Russo <diego.russo@arm.com>2022-05-30 13:34:14 +0100
commit0efca3cadbad5517a59884576ddb90cfe7ac30f8 (patch)
treeabed6cb6fbf3c439fc8d947f505b6a53d5daeb1e /src/mlia/__init__.py
parent0777092695c143c3a54680b5748287d40c914c35 (diff)
downloadmlia-0efca3cadbad5517a59884576ddb90cfe7ac30f8.tar.gz
Add MLIA codebase0.3.0-rc.1
Add MLIA codebase including sources and tests. Change-Id: Id41707559bd721edd114793618d12ccd188d8dbd
Diffstat (limited to 'src/mlia/__init__.py')
-rw-r--r--src/mlia/__init__.py22
1 files changed, 22 insertions, 0 deletions
diff --git a/src/mlia/__init__.py b/src/mlia/__init__.py
new file mode 100644
index 0000000..ed9ae87
--- /dev/null
+++ b/src/mlia/__init__.py
@@ -0,0 +1,22 @@
+# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""Init of MLIA."""
+import logging
+import os
+
+import pkg_resources
+
+# 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__ = pkg_resources.get_distribution("mlia").version