Changing single to double quotes for quote consistency

This commit is contained in:
MalikMAlna 2023-04-06 20:07:08 -04:00
parent 334f36d844
commit 17fb6f668a

View File

@ -67,7 +67,7 @@ def load_data(config, tokenizer):
dataset = load_dataset("json", data_files=files, split="train") dataset = load_dataset("json", data_files=files, split="train")
else: else:
dataset = load_dataset(dataset_path, split='train') dataset = load_dataset(dataset_path, split="train")
dataset = dataset.train_test_split(test_size=.05, seed=config["seed"]) dataset = dataset.train_test_split(test_size=.05, seed=config["seed"])