data.py

ds = load_dataset("codeparrot/github-code", streaming=True, split="train")
print(next(iter(ds)))