diff options
-rw-r--r-- | test/test_smooth.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_smooth.py b/test/test_smooth.py index 7aceb42..2afa4f8 100644 --- a/test/test_smooth.py +++ b/test/test_smooth.py @@ -20,7 +20,7 @@ def test_smooth_ot_dual(): M = ot.dist(x, x) - with pytest.raises(NotImplementedError) as e_info: + with pytest.raises(NotImplementedError): Gl2, log = ot.smooth.smooth_ot_dual(u, u, M, 1, reg_type='none') Gl2, log = ot.smooth.smooth_ot_dual(u, u, M, 1, reg_type='l2', log=True, stopThr=1e-10) @@ -55,7 +55,7 @@ def test_smooth_ot_semi_dual(): M = ot.dist(x, x) - with pytest.raises(NotImplementedError) as e_info: + with pytest.raises(NotImplementedError): Gl2, log = ot.smooth.smooth_ot_semi_dual(u, u, M, 1, reg_type='none') Gl2, log = ot.smooth.smooth_ot_semi_dual(u, u, M, 1, reg_type='l2', log=True, stopThr=1e-10) |