explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-15 10:42:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix tests for gpu

Browse Source
This commit is contained in:
Matthw Honnibal 2019-10-27 22:19:18 +01:00
parent 165e378082
commit 426b745640
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/tests/test_misc.py
View File

@ -95,10 +95,16 @@ def test_PrecomputableAffine(nO=4, nI=5, nF=3, nP=2):
def test_prefer_gpu(): def test_prefer_gpu():
try:
import cupy
except ImportError:
assert not prefer_gpu() assert not prefer_gpu()
def test_require_gpu(): def test_require_gpu():
try:
import cupy
except ImportError:
with pytest.raises(ValueError): with pytest.raises(ValueError):
require_gpu() require_gpu()