From c8ff6d95f059f10e2fe5d07a7bfb3f91c3408161 Mon Sep 17 00:00:00 2001 From: Pierre Chapuis Date: Wed, 28 Aug 2024 10:39:41 +0200 Subject: [PATCH] fix deprecation warning --- src/refiners/training_utils/config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/refiners/training_utils/config.py b/src/refiners/training_utils/config.py index 7b59415..57921e2 100644 --- a/src/refiners/training_utils/config.py +++ b/src/refiners/training_utils/config.py @@ -1,5 +1,5 @@ from enum import Enum -from logging import warn +from logging import warning from pathlib import Path from typing import Annotated, Any, Callable, Iterable, Literal, Type, TypeVar @@ -132,7 +132,7 @@ class OptimizerConfig(BaseModel): ) case Optimizers.Prodigy: if self.learning_rate != 1.0: - warn("Prodigy learning rate is not 1.0, this might cause instability.") + warning("Prodigy learning rate is not 1.0, this might cause instability.") return Prodigy( lr=self.learning_rate, params=params,