Update test_mlp.py

added make_rng