Skip to content

Commit

Permalink
[fix] change print to logger info
Browse files Browse the repository at this point in the history
  • Loading branch information
YizhenJia committed Feb 1, 2025
1 parent d628561 commit bf59cec
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/lmflow/datasets/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,7 @@ def __init__(self, data_args: DatasetArguments=None, backend: str="huggingface",
# check if the dataset is in the correct format and get the dataset type (text_only, text2text, etc.)
self._check_hf_json_format(data_files)
# Load the dataset using the HuggingFace dataset library
print('loading datasets')
logger.info('Loading datasets')
extensions = "json"
raw_dataset = load_dataset(
extensions,
Expand Down

0 comments on commit bf59cec

Please sign in to comment.