Bug in MnistClassifier.py

[Maximophone]

def predict(self,X,features_names):
        assert X.shape[0]>=1, 'wrong shape 0'
        if X.shape[0]==784:
                  X = X.reshape(1,28,28,1)
        else:
                  X = X.reshape(X.shape[0],28,28,1)

What if I want to make a prediction on a batch of 784 images? As a general guideline, the interface methods can check the format of what they receive but should not assume different formats and implement different behaviours accordingly.