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

Fix Stepwisestate'

This commit is contained in:
Matthew Honnibal 2017-04-15 13:35:01 +02:00
parent 0739ae7b76
commit 1a98e48b8e
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/syntax/parser.pyx
View File

@ -383,7 +383,7 @@ cdef class StepwiseState:
def __init__(self, Parser parser, Doc doc, GoldParse gold=None): def __init__(self, Parser parser, Doc doc, GoldParse gold=None):
self.parser = parser self.parser = parser
self.doc = doc self.doc = doc
if gold: if gold is not None:
self.gold = gold self.gold = gold
else: else:
self.gold = GoldParse(doc) self.gold = GoldParse(doc)
@ -427,6 +427,8 @@ cdef class StepwiseState:
""" """
Find the action-costs for the current state. Find the action-costs for the current state.
""" """
if not self.gold:
raise ValueError("Can't set costs: No GoldParse provided")
self.parser.moves.set_costs(self.eg.c.is_valid, self.eg.c.costs, self.parser.moves.set_costs(self.eg.c.is_valid, self.eg.c.costs,
self.stcls, self.gold) self.stcls, self.gold)
costs = {} costs = {}