[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-06-27 12:20:08 +00:00
commit 966c00aa58

View file

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