Update datasets.py

added make_rng