Skip to content

Commit b668627

Browse files
authored
update arguments (#1044)
1 parent 4e18bff commit b668627

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

swift/llm/utils/argument.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -276,11 +276,11 @@ def _handle_dataset_sample(self):
276276
return
277277
_dataset = self.dataset[0]
278278
train_sample = parse_dataset_name(_dataset)[3]
279-
if train_sample is None:
279+
if train_sample == -1:
280280
train_sample = self.train_dataset_sample
281281
elif self.train_dataset_sample < train_sample:
282+
_dataset = _dataset[:_dataset.find('#')]
282283
train_sample = self.train_dataset_sample
283-
_dataset = _dataset[:_dataset.find('#')]
284284
_dataset = f'{_dataset}#{train_sample}'
285285
self.dataset[0] = _dataset
286286
self.train_dataset_sample = -1

0 commit comments

Comments
 (0)