1 files changed, 13 insertions, 0 deletions
diff --git a/tests/mlia/utils/logging.py b/tests/mlia/utils/logging.py
new file mode 100644
index 0000000..d223fb2
--- /dev/null
+++ b/tests/mlia/utils/logging.py
@@ -0,0 +1,13 @@
+# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""Utils for logging."""
+import logging
+
+
+def clear_loggers() -> None:
+    """Close the log handlers."""
+    for _, logger in logging.Logger.manager.loggerDict.items():
+        if not isinstance(logger, logging.PlaceHolder):
+            for handler in logger.handlers:
+                handler.close()
+                logger.removeHandler(handler)