From f5fbc82f2bd65b9f19d8b7cb2b3702ef80da9714 Mon Sep 17 00:00:00 2001 From: JZacaroli <32681029+JZacaroli@users.noreply.github.com> Date: Thu, 12 Jun 2025 18:23:31 +0100 Subject: [PATCH] Fix logging import in evaluate.py (#2782) (#2783) * Fix logging import in evaluate.py (#2782) * chore: lint --------- Co-authored-by: Joe Zacaroli Co-authored-by: Wing Lian --- src/axolotl/evaluate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/evaluate.py b/src/axolotl/evaluate.py index 6d6813730..2b5869939 100644 --- a/src/axolotl/evaluate.py +++ b/src/axolotl/evaluate.py @@ -7,7 +7,6 @@ from pathlib import Path from typing import Dict, Optional import torch -from accelerate.logging import get_logger from datasets import Dataset from transformers.trainer import Trainer @@ -17,6 +16,7 @@ from axolotl.train import ( ) from axolotl.utils.dict import DictDefault from axolotl.utils.distributed import cleanup_distributed +from axolotl.utils.logging import get_logger from axolotl.utils.trainer import setup_trainer project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))