zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-06 17:04:34 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix pre-commit lint

Browse Source
This commit is contained in:
madroid
2024-09-29 14:22:38 +08:00
parent 32a0bc903a
commit a83ca489a4
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
llms/mlx_lm/tuner/datasets.py
View File

@@ -107,7 +107,7 @@ def load_local_dataset(data_path: Path, tokenizer: PreTrainedTokenizer):
def load_hf_dataset(data_id: str, tokenizer: PreTrainedTokenizer): def load_hf_dataset(data_id: str, tokenizer: PreTrainedTokenizer):
from datasets import load_dataset, exceptions from datasets import exceptions, load_dataset
try: try:
dataset = load_dataset(data_id) dataset = load_dataset(data_id)
@@ -120,9 +120,7 @@ def load_hf_dataset(data_id: str, tokenizer: PreTrainedTokenizer):
] ]
except exceptions.DatasetNotFoundError: except exceptions.DatasetNotFoundError:
raise ValueError( raise ValueError(f"Not found Hugging Face dataset: {data_id} .")
f"Not found Hugging Face dataset: {data_id} ."
)
return train, valid, test return train, valid, test