From 25e69f2d36ca4212a26ae9f828e338bc15c1e9e2 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Thu, 20 Nov 2025 12:57:53 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 unsloth-cli.py | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/unsloth-cli.py b/unsloth-cli.py
index 44232e19d7..79b3ef5296 100644
--- a/unsloth-cli.py
+++ b/unsloth-cli.py
@@ -101,7 +101,7 @@ def run(args):
 
     def load_dataset_smart(args):
         from transformers.utils import strtobool
-        
+
         if args.raw_text_file:
             # Use raw text loader
             loader = RawTextDataLoader(tokenizer, args.chunk_size, args.stride)
@@ -112,16 +112,19 @@ def run(args):
             dataset = loader.load_from_file(args.dataset)
         else:
             # Check for modelscope usage
-            use_modelscope = strtobool(os.environ.get("UNSLOTH_USE_MODELSCOPE", "False"))
+            use_modelscope = strtobool(
+                os.environ.get("UNSLOTH_USE_MODELSCOPE", "False")
+            )
             if use_modelscope:
                 from modelscope import MsDataset
-                dataset = MsDataset.load(args.dataset, split="train")
+
+                dataset = MsDataset.load(args.dataset, split = "train")
             else:
                 # Existing HuggingFace dataset logic
-                dataset = load_dataset(args.dataset, split="train")
-            
+                dataset = load_dataset(args.dataset, split = "train")
+
             # Apply formatting for structured datasets
-            dataset = dataset.map(formatting_prompts_func, batched=True)
+            dataset = dataset.map(formatting_prompts_func, batched = True)
         return dataset
 
     # Load dataset using smart loader