Adding logging enhancement

2023-07-14 07:26:19 -05:00
parent ef17e15483
commit 553a86b52c
14 changed files with 119 additions and 73 deletions
--- a/tests/test_prompt_tokenizers.py
+++ b/tests/test_prompt_tokenizers.py
@@ -16,8 +16,11 @@ from axolotl.prompt_tokenizers import (
    ShareGPTPromptTokenizingStrategy,
 )
 from axolotl.prompters import AlpacaPrompter, PromptStyle, ShareGPTPrompter
+from axolotl.logging_config import configure_logging

-logging.basicConfig(level="INFO")
+configure_logging()
+
+LOG = logging.getLogger("axolotl")


 class TestPromptTokenizationStrategies(unittest.TestCase):