From 9300f8de65de7ab86a4ba22b23a13403a593c369 Mon Sep 17 00:00:00 2001 From: suluyana Date: Thu, 17 Oct 2024 11:16:40 +0800 Subject: [PATCH] fix lint --- modelscope/msdatasets/context/dataset_context_config.py | 3 ++- tests/msdatasets/test_general_datasets.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/modelscope/msdatasets/context/dataset_context_config.py b/modelscope/msdatasets/context/dataset_context_config.py index bfe7dbd1..fce262b0 100644 --- a/modelscope/msdatasets/context/dataset_context_config.py +++ b/modelscope/msdatasets/context/dataset_context_config.py @@ -17,7 +17,8 @@ class DatasetContextConfig: data_files: Union[str, Sequence[str], Mapping[str, Union[str, Sequence[str]]]], download_mode: DownloadMode, cache_root_dir: str, - use_streaming: bool, stream_batch_size: int, trust_remote_code: bool, **kwargs): + use_streaming: bool, stream_batch_size: int, + trust_remote_code: bool, **kwargs): self._download_config = None self._data_meta_config = None diff --git a/tests/msdatasets/test_general_datasets.py b/tests/msdatasets/test_general_datasets.py index 0792cf46..1fd96f8f 100644 --- a/tests/msdatasets/test_general_datasets.py +++ b/tests/msdatasets/test_general_datasets.py @@ -121,7 +121,8 @@ class GeneralMsDatasetTest(unittest.TestCase): f'please check the url: {py_script_url}' # Load the dataset - ds = MsDataset.load(py_script_file, subset_name='cola', split='train') + ds = MsDataset.load( + py_script_file, subset_name='cola', split='train') sample = next(iter(ds)) logger.info(f'>>output of test_local_py_script:\n {sample}') assert sample