From 966c00aa587dda3f42a843060fc2579dbcf94ad6 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sat, 27 Jun 2026 12:20:08 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- unsloth/save.py | 1 - 1 file changed, 1 deletion(-) diff --git a/unsloth/save.py b/unsloth/save.py index beb35135c9..e116932a60 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -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())