From 70be03461db45de50ecd073b9795093ead1ba5f5 Mon Sep 17 00:00:00 2001
From: Slasnista <stan.chambon@gmail.com>
Date: Fri, 4 Aug 2017 11:16:30 +0200
Subject: added test for fit_transform + correction of fit_transform bug
 (missing return self)

---
 ot/da.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'ot/da.py')

diff --git a/ot/da.py b/ot/da.py
index fb2fd36..80649a7 100644
--- a/ot/da.py
+++ b/ot/da.py
@@ -1283,6 +1283,8 @@ class SinkhornTransport(BaseTransport):
             numItermax=self.max_iter, stopThr=self.tol,
             verbose=self.verbose, log=self.log)
 
+        return self
+
 
 class EMDTransport(BaseTransport):
     """Domain Adapatation OT method based on Earth Mover's Distance
@@ -1357,3 +1359,5 @@ class EMDTransport(BaseTransport):
             # verbose=self.verbose,
             # log=self.log
         )
+
+        return self
-- 
cgit v1.2.3