scikit-learn
diff --git a/‎sklearn/neighbors/_nca.py
Copy file name to clipboardExpand all lines: sklearn/neighbors/_nca.py
+1-1Lines changed: 1 addition & 1 deletion b/‎sklearn/neighbors/_nca.py
Copy file name to clipboardExpand all lines: sklearn/neighbors/_nca.py
+1-1Lines changed: 1 addition & 1 deletion
@@ -229,7 +229,7 @@ def fit(self, X, y):
         # (n_samples, n_samples)
 
         # Initialize the transformation
-        transformation = self._initialize(X, y, init)
+        transformation = np.ravel(self._initialize(X, y, init))
 
         # Create a dictionary of parameters to be passed to the optimizer
         disp = self.verbose - 2 if self.verbose > 1 else -1