explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-05 12:50:20 +03:00
Code Issues Packages Projects Releases Wiki Activity

make flake8 happy again

Browse Source
This commit is contained in:
svlandeg 2022-01-20 17:20:34 +01:00
parent 4d9d9c5a28
commit ca6aa239bc
1 changed files with 27 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
spacy/tests/test_misc.py
View File

@ -78,33 +78,33 @@ def test_util_get_package_path(package):
assert isinstance(path, Path) assert isinstance(path, Path)
@pytest.mark.xfail(reason="No precomputable affine") # @pytest.mark.skip(reason="No precomputable affine")
def test_PrecomputableAffine(nO=4, nI=5, nF=3, nP=2): # def test_PrecomputableAffine(nO=4, nI=5, nF=3, nP=2):
model = PrecomputableAffine(nO=nO, nI=nI, nF=nF, nP=nP).initialize() # model = PrecomputableAffine(nO=nO, nI=nI, nF=nF, nP=nP).initialize()
assert model.get_param("W").shape == (nF, nO, nP, nI) # assert model.get_param("W").shape == (nF, nO, nP, nI)
tensor = model.ops.alloc((10, nI)) # tensor = model.ops.alloc((10, nI))
Y, get_dX = model.begin_update(tensor) # Y, get_dX = model.begin_update(tensor)
assert Y.shape == (tensor.shape[0] + 1, nF, nO, nP) # assert Y.shape == (tensor.shape[0] + 1, nF, nO, nP)
dY = model.ops.alloc((15, nO, nP)) # dY = model.ops.alloc((15, nO, nP))
ids = model.ops.alloc((15, nF)) # ids = model.ops.alloc((15, nF))
ids[1, 2] = -1 # ids[1, 2] = -1
dY[1] = 1 # dY[1] = 1
assert not model.has_grad("pad") # assert not model.has_grad("pad")
d_pad = _backprop_precomputable_affine_padding(model, dY, ids) # d_pad = _backprop_precomputable_affine_padding(model, dY, ids)
assert d_pad[0, 2, 0, 0] == 1.0 # assert d_pad[0, 2, 0, 0] == 1.0
ids.fill(0.0) # ids.fill(0.0)
dY.fill(0.0) # dY.fill(0.0)
dY[0] = 0 # dY[0] = 0
ids[1, 2] = 0 # ids[1, 2] = 0
ids[1, 1] = -1 # ids[1, 1] = -1
ids[1, 0] = -1 # ids[1, 0] = -1
dY[1] = 1 # dY[1] = 1
ids[2, 0] = -1 # ids[2, 0] = -1
dY[2] = 5 # dY[2] = 5
d_pad = _backprop_precomputable_affine_padding(model, dY, ids) # d_pad = _backprop_precomputable_affine_padding(model, dY, ids)
assert d_pad[0, 0, 0, 0] == 6 # assert d_pad[0, 0, 0, 0] == 6
assert d_pad[0, 1, 0, 0] == 1 # assert d_pad[0, 1, 0, 0] == 1
assert d_pad[0, 2, 0, 0] == 0 # assert d_pad[0, 2, 0, 0] == 0
def test_prefer_gpu(): def test_prefer_gpu():