Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix bug where huffman codec died if given empty freqs dict.

This commit is contained in:
Matthew Honnibal 2016-10-23 16:28:05 +02:00
parent 99ff8b902f
commit 49c117960c
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
spacy/serialize/huffman.pyx
View File

@ -46,12 +46,14 @@ cdef class HuffmanCodec:
item.first = item1.first + item2.first
item.second = self.nodes.size()-1
queue.push(item)
item = queue.top()
self.root = self.nodes[item.second]
# Careful of empty freqs dicts
cdef Code path
path.bits = 0
path.length = 0
assign_codes(self.nodes, self.codes, item.second, path)
if queue.size() >= 1:
item = queue.top()
self.root = self.nodes[item.second]
path.bits = 0
path.length = 0
assign_codes(self.nodes, self.codes, item.second, path)
def encode(self, msg, BitArray bits=None):
if bits is None: