mnist, #deep learing, #pickle error, #numpy version, #ram

I want to run this code
https://github.com/oreilly-japan/deep-learning-from-scratch/blob/master/dataset/mnist.py
but there are sth wrong that numpy version is not suitable and Ipad pro's ram is insufficient

so I modified code that refere http://poipoides.hatenablog.com/entry/2017/02/07/212013

however there is still minor error

init_mnist() missing 1 required positional argument: 'key (lin 171)

I can't fix it Plz help me
I want use pythonista smartly

https://github.com/oreilly-japan/deep-learning-from-scratch/blob/master/dataset/mnist.py

this is code that I use