added test for fit_transform + correction of fit_transform bug (missing return self)

1 files changed, 4 insertions, 0 deletions

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