diff options
-rw-r--r-- | ot/da.py | 2 | ||||
-rw-r--r-- | ot/optim.py | 2 |
2 files changed, 2 insertions, 2 deletions
@@ -2,7 +2,7 @@ domain adaptation with optimal transport """ import numpy as np -from bregman import sinkhorn +from .bregman import sinkhorn diff --git a/ot/optim.py b/ot/optim.py index 632eac1..1afbea3 100644 --- a/ot/optim.py +++ b/ot/optim.py @@ -6,7 +6,7 @@ Optimization algorithms for OT import numpy as np import scipy as sp from scipy.optimize.linesearch import scalar_search_armijo -from lp import emd +from .lp import emd # The corresponding scipy function does not work for matrices def line_search_armijo(f,xk,pk,gfk,old_fval,args=(),c1=1e-4,alpha0=0.99): |