Commit b2598de8 authored by lintangsutawika's avatar lintangsutawika
Browse files

process int label to string for greedy_until

parent 8da6033f
......@@ -978,6 +978,9 @@ class ConfigurableTask(Task):
elif self.OUTPUT_TYPE == "greedy_until":
gold = self.doc_to_target(doc)
if type(gold) == int:
choices = self.doc_to_choice(doc)
gold = choices[gold]
for key, result in zip(self._metric_fn_list.keys(), results):
if self.multiple_target:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment