Skip to content

Commit

Permalink
fix deprecation warning
Browse files Browse the repository at this point in the history
  • Loading branch information
catwell committed Aug 28, 2024
1 parent e643006 commit c8ff6d9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/refiners/training_utils/config.py
Original file line number Diff line number Diff line change
@@ -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

Expand Down Expand Up @@ -132,7 +132,7 @@ def get(self, params: ParamsT) -> Optimizer:
)
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,
Expand Down

0 comments on commit c8ff6d9

Please sign in to comment.