diff options
author | Nathan Cassereau <84033440+ncassereau-idris@users.noreply.github.com> | 2021-12-03 12:37:05 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-12-03 12:37:05 +0100 |
commit | ca69658400dc2ef6a7d3e531acffcd107400085f (patch) | |
tree | b77a28821067be5240cec2082fa1f119b1cfd1cd /test/test_bregman.py | |
parent | cb510644b2fd65e4ce216a7799ce7401f71548b8 (diff) |
[MRG] Cupy backend (#315)
* Cupy backend
* pep8 + bug
* working even if cupy not installed
* attempt to force codecov to ignore cupy because no gpu can be used for testing on github
* docstring
* readme
Diffstat (limited to 'test/test_bregman.py')
-rw-r--r-- | test/test_bregman.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/test/test_bregman.py b/test/test_bregman.py index 830052d..f42ac6f 100644 --- a/test/test_bregman.py +++ b/test/test_bregman.py @@ -888,6 +888,7 @@ def test_implemented_methods(): ot.bregman.sinkhorn2(a, b, M, epsilon, method=method) +@pytest.skip_backend("cupy") @pytest.skip_backend("jax") @pytest.mark.filterwarnings("ignore:Bottleneck") def test_screenkhorn(nx): |