data.py

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