From 077a54d2b1fedfc94b0fd588a36131438a073640 Mon Sep 17 00:00:00 2001
From: NanoCode012 <nano@axolotl.ai>
Date: Mon, 3 Feb 2025 16:03:41 +0700
Subject: [PATCH] fix: deprecate old types

---
 src/axolotl/core/trainer_builder.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py
index c9cc2070e..325c02ef3 100755
--- a/src/axolotl/core/trainer_builder.py
+++ b/src/axolotl/core/trainer_builder.py
@@ -25,7 +25,7 @@ import os
 import sys
 from abc import abstractmethod
 from pathlib import Path
-from typing import List, Type, Union
+from typing import Type, Union
 
 import torch
 import transformers
@@ -155,7 +155,7 @@ class TrainerBuilderBase(abc.ABC):
     def build(self, total_num_steps):
         pass
 
-    def get_callbacks(self) -> List[TrainerCallback]:
+    def get_callbacks(self) -> list[TrainerCallback]:
         callbacks = []
 
         plugin_manager = PluginManager.get_instance()