Fix indexing in MDM

pyRiemann · Jul 30, 2015 · 02f47c5 · 02f47c5
1 parent 417d0f7
commit 02f47c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyriemann/classification.py b/pyriemann/classification.py
@@ -41,9 +41,9 @@ def fit(self, X, y):
         if self.n_jobs == 1:
             for l in self.classes:
                 self.covmeans.append(
-                    mean_covariance(X[y == l, :, :], metric=self.metric_mean))
+                    mean_covariance(X[y == l], metric=self.metric_mean))
         else:
-            self.covmeans = Parallel(n_jobs=self.n_jobs)(delayed(mean_covariance)(X[y == l, :, :],metric=self.metric_mean) for l in self.classes)
+            self.covmeans = Parallel(n_jobs=self.n_jobs)(delayed(mean_covariance)(X[y == l],metric=self.metric_mean) for l in self.classes)
 
         return self