warning, not warn

mozilla-ai · Apr 9, 2024 · 2fb8a05 · 2fb8a05
1 parent 2b65088
commit 2fb8a05
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/lm_buddy/jobs/evaluation/prometheus.py b/src/lm_buddy/jobs/evaluation/prometheus.py
@@ -100,7 +100,7 @@ def get_response_with_retries(
             feedback, score = parse_response(config, response)
             break
         except (OpenAIError, BadResponseError) as e:
-            logger.warn(
+            logger.warning(
                 f"{e.message}: "
                 f"Retrying ({current_retry_attempt}/{config.evaluation.max_retries})"
             )
@@ -144,7 +144,7 @@ def data_generator():
             # skip those examples which are too long
             tokenized_prompt = tokenizer(prompt, truncation=False)
             if len(tokenized_prompt["input_ids"]) > 3072:
-                logger.warn(f"Skipping row due to prompt exceeding token limit: {prompt=}")
+                logger.warning(f"Skipping row due to prompt exceeding token limit: {prompt=}")
                 continue
 
             # prepare output