summaryrefslogtreecommitdiff
path: root/test/test_da.py
diff options
context:
space:
mode:
authorievred <ievgen.redko@univ-st-etienne.fr>2020-04-01 09:49:24 +0200
committerievred <ievgen.redko@univ-st-etienne.fr>2020-04-01 09:49:24 +0200
commit6b8477d1c08696a08a1b71642712d83e560f9623 (patch)
tree702f920a75bf3f9c9b316d8c21cc71211ff27f28 /test/test_da.py
parentb1f87363b160735b6e2df59380f9de56b7934b53 (diff)
pep8
Diffstat (limited to 'test/test_da.py')
-rw-r--r--test/test_da.py7
1 files changed, 3 insertions, 4 deletions
diff --git a/test/test_da.py b/test/test_da.py
index a13550c..f700df9 100644
--- a/test/test_da.py
+++ b/test/test_da.py
@@ -511,7 +511,6 @@ def test_mapping_transport_class():
def test_linear_mapping():
-
ns = 150
nt = 200
@@ -529,7 +528,6 @@ def test_linear_mapping():
def test_linear_mapping_class():
-
ns = 150
nt = 200
@@ -568,7 +566,7 @@ def test_jcpot_transport_class():
Xs = [Xs1, Xs2]
ys = [ys1, ys2]
- otda = ot.da.JCPOTTransport(reg_e=0.01, max_iter=1000, tol=1e-9, verbose=True, log = True)
+ otda = ot.da.JCPOTTransport(reg_e=0.01, max_iter=1000, tol=1e-9, verbose=True, log=True)
# test its computed
otda.fit(Xs=Xs, ys=ys, Xt=Xt)
@@ -592,7 +590,8 @@ def test_jcpot_transport_class():
# test margin constraints w.r.t. modified source weights for each source domain
assert_allclose(
- np.dot(otda.log_['all_domains'][i]['D1'], np.sum(otda.coupling_[i], axis=1)), otda.proportions_, rtol=1e-3, atol=1e-3)
+ np.dot(otda.log_['all_domains'][i]['D1'], np.sum(otda.coupling_[i], axis=1)), otda.proportions_, rtol=1e-3,
+ atol=1e-3)
# test transform
transp_Xs = otda.transform(Xs=Xs)