From 8ee4b02738e8ff808af3b548918e5eefe2fa6b9a Mon Sep 17 00:00:00 2001
From: Maximilian Schmidt <maximilian.schmidt@ims.uni-stuttgart.de>
Date: Mon, 4 Mar 2024 18:30:13 +0100
Subject: [PATCH] Adding console handler for logging
---
utils.py | 13 +++++++++++++
1 file changed, 13 insertions(+)
diff --git a/utils.py b/utils.py
index fd3cd67..cbcf4a1 100644
--- a/utils.py
+++ b/utils.py
@@ -16,6 +16,17 @@ current_directory = Path(__file__).resolve().parent
logger = logging.getLogger("test-classifier")
logger.setLevel(level=logging.DEBUG)
+# create console handler and set level to debug
+console_handler = logging.StreamHandler()
+console_handler.setLevel(logging.DEBUG)
+# formatter for console handler
+console_formatter = logging.Formatter(
+ "%(asctime)s - %(name)s - %(levelname)s: %(message)s"
+)
+console_handler.setFormatter(console_formatter)
+# add console handler to logger
+logger.addHandler(console_handler)
+
run_name_prompt = """
Create a random name that sounds german or dutch
The parts should be separated by underscores and contain only lowercase.
@@ -33,10 +44,12 @@ def initialize_run_directory(model: OpenAI | Llama2):
run_directory = current_directory / f"runs/{run_name}"
# TODO what if name exists?
run_directory.mkdir(parents=True, exist_ok=False)
+ # create file handler and set level to debug
file_handler = logging.FileHandler(run_directory / "output.log")
file_handler.setLevel(logging.DEBUG)
formatter = logging.Formatter("%(asctime)s %(levelname)s %(message)s")
file_handler.setFormatter(formatter)
+ # add file handler to logger
logger.addHandler(file_handler)
logger.info(f"Hello my name is {run_name} and I live in {run_directory}")
--
GitLab