explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-25 01:16:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix #1518: vocab.vectors.resize() didn't work

Browse Source
This commit is contained in:
Matthew Honnibal 2017-11-08 22:18:37 +01:00
parent dec79ca332
commit a5ea0fdf5a
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spacy/tests/regression/test_issue1518.py Normal file
View File

@ -0,0 +1,7 @@
from ...vectors import Vectors
def test_issue1518():
'''Test vectors.resize() works.'''
vectors = Vectors(shape=(10, 10))
vectors.add(u'hello', row=2)
vectors.resize((5, 9))

2
spacy/vectors.pyx
View File

@ -150,7 +150,7 @@ cdef class Vectors:
filled = {row for row in self.key2row.values()} filled = {row for row in self.key2row.values()}
self._unset = {row for row in range(shape[0]) if row not in filled} self._unset = {row for row in range(shape[0]) if row not in filled}
removed_items = [] removed_items = []
for key, row in dict(self.key2row.items()): for key, row in self.key2row.items():
if row >= shape[0]: if row >= shape[0]:
self.key2row.pop(key) self.key2row.pop(key)
removed_items.append((key, row)) removed_items.append((key, row))