[fix] push fix by linter

2026-07-13 00:50:06 +08:00 · 2023-01-06 17:14:19 +00:00
parent 577b14a702
commit 50e7472ae6
1 changed files with 0 additions and 1 deletions
@@ -5,7 +5,6 @@ from typing import Any, Dict, List, Optional, Tuple, Union

 import torch
 from torch import nn
-from torch.utils.data import Dataset
 from transformers import PreTrainedModel, Trainer, TrainingArguments, get_cosine_schedule_with_warmup
 from utils import get_dataset, get_loss, get_model, get_tokenizer, read_yamls