Fix sklearn wrapper unit test in Python 3?

Author: fchollet

keras/wrappers/scikit_learn.py

@@ -320,7 +320,8 @@ def predict(self, x, **kwargs):
                 Predictions.
         """
         kwargs = self.filter_sk_params(Sequential.predict, kwargs)
-        return np.squeeze(self.model.predict(x, **kwargs), axis=-1)
+        preds = self.model.predict(x, **kwargs)
+        return np.squeeze(preds, axis=len(preds.shape) - 1)
 
     def score(self, x, y, **kwargs):
         """Returns the mean loss on the given test data and labels.