Refactor logger imports

ls1intum · Sep 11, 2024 · 93d9bb7 · 93d9bb7
1 parent e1ac199
commit 93d9bb7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/modules/modeling/module_modeling_llm/module_modeling_llm/generate_suggestions.py b/modules/modeling/module_modeling_llm/module_modeling_llm/generate_suggestions.py
@@ -5,7 +5,6 @@
 from pydantic import BaseModel, Field
 
 from athena import emit_meta
-from athena.logger import logger
 from athena.modeling import Exercise, Submission, Feedback
 from module_modeling_llm.config import BasicApproachConfig
 from langchain_core.output_parsers import PydanticOutputParser

diff --git a/modules/modeling/module_modeling_llm/module_modeling_llm/helpers/llm_utils.py b/modules/modeling/module_modeling_llm/module_modeling_llm/helpers/llm_utils.py
@@ -4,7 +4,7 @@
 from langchain_core.pydantic_v1 import BaseModel, ValidationError
 from langchain_core.runnables import RunnableSequence
 from athena import get_experiment_environment
-from assessment_module_manager.logger import logger
+from athena.logger import logger
 
 T = TypeVar("T", bound=BaseModel)
 
@@ -46,7 +46,7 @@ async def predict_and_parse(
     try:
         return await chain.ainvoke(prompt_input, config={"tags": tags})
     except (ValidationError) as e:
-        
+
         logger.error("Exception type: %s, Message: %s", type(e).__name__, e)
 
         return None