Update mnist_logreg.py

Removed unnecessary import