diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/test_da.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/test_da.py b/test/test_da.py index 372ebd4..4eaf193 100644 --- a/test/test_da.py +++ b/test/test_da.py @@ -589,7 +589,7 @@ 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, + np.dot(otda.log_['D1'][i], np.sum(otda.coupling_[i], axis=1)), otda.proportions_, rtol=1e-3, atol=1e-3) # test transform |