fix: deprecate old types

This commit is contained in:
NanoCode012
2025-02-03 16:03:41 +07:00
parent 053e5fd7d1
commit 077a54d2b1

View File

@@ -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()