Commit d99e6cf4 authored by lintangsutawika's avatar lintangsutawika
Browse files

fix doc_to_choice being called when target is digit

parent 3e9c9a95
......@@ -1002,9 +1002,13 @@ class ConfigurableTask(Task):
elif self.OUTPUT_TYPE == "greedy_until":
gold = self.doc_to_target(doc)
if type(gold) == int:
if self._config.doc_to_choice is not None:
# If you set doc_to_choice,
# it assumes that doc_to_target returns a number.
choices = self.doc_to_choice(doc)
gold = choices[gold]
else:
gold = str(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