Update datasets.py

Added make_rng