Hi,
this is possible but not nearly as clean as the approach which involves converting data to JSON Lines and splitting it to have one file per split:
import datasets
ddict = datasets.DatasetDict()
for split in ["train", "test", "valid"]:
   ddict.update(datasets.load_dataset("json", data_files={split: "path/to/data/file.json"}, field=split)