Fix to BaseTransport.transform_labels() (#208)

* Fix to BaseTransport.transform_labels()

Issue #207

* Fix - forgot to commit

Author: samarth4149

ot/da.py

@@ -1111,7 +1111,7 @@ def transform_labels(self, ys=None):
             D1 = np.zeros((n, len(ysTemp)))
 
             # perform label propagation
-            transp = self.coupling_ / np.sum(self.coupling_, 1)[:, None]
+            transp = self.coupling_ / np.sum(self.coupling_, 0, keepdims=True)
 
             # set nans to 0
             transp[~ np.isfinite(transp)] = 0