From 48213b868e78ac66028704203d40e8390ad1fa93 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=8D=97=E5=B7=9D?= Date: Mon, 6 Nov 2023 14:07:10 +0800 Subject: [PATCH] fixed typo: separate --- .gitignore | 1 + finetune/utils/data/data_utils.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.gitignore b/.gitignore index bb7a4da5..40b8311e 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,4 @@ +.idea/* .vscode/* # Local History for Visual Studio Code diff --git a/finetune/utils/data/data_utils.py b/finetune/utils/data/data_utils.py index 34394395..a7e19aec 100644 --- a/finetune/utils/data/data_utils.py +++ b/finetune/utils/data/data_utils.py @@ -452,7 +452,7 @@ def __init__(self, max_size, small_batch_size): self.max_size = max_size self.small_batch_size = small_batch_size - def seperate(self): + def separate(self): small_dataset = [] for large_batch in self.dataset: if type(large_batch) == list or type(large_batch) == tuple: @@ -483,7 +483,7 @@ def add(self, data): if len(self.dataset) < self.max_size: self.dataset.append(data) if len(self.dataset) == self.max_size: - return self.seperate() + return self.separate() else: return None else: