[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
a1b4b9d431
commit
966c00aa58
1 changed files with 0 additions and 1 deletions
|
|
@ -3933,7 +3933,6 @@ def _unsloth_save_compressed_tensors(
|
|||
# row subsample below applies and we do not save every split to the temp dir.
|
||||
try:
|
||||
from datasets import DatasetDict
|
||||
|
||||
if isinstance(ds_to_save, DatasetDict):
|
||||
ds_to_save = ds_to_save.get("train", None) or next(
|
||||
iter(ds_to_save.values())
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue