From f3ecbf9c6fc6a21dd519c9f62127a18bb5bff029 Mon Sep 17 00:00:00 2001 From: Tony Lekhtman Date: Thu, 18 Jun 2020 17:58:16 +0300 Subject: [PATCH] Update hub_utils.py fix bug for print_alignment --- fairseq/hub_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fairseq/hub_utils.py b/fairseq/hub_utils.py index 1b1091c881..c249eb23f5 100644 --- a/fairseq/hub_utils.py +++ b/fairseq/hub_utils.py @@ -189,7 +189,7 @@ def getarg(name, default): )) if hypo['alignment'] is not None and getarg('print_alignment', False): logger.info('A\t{}'.format( - ' '.join(map(lambda x: str(utils.item(x)), hypo['alignment'].int().cpu())) + ' '.join(['{}-{}'.format(src_idx, tgt_idx) for src_idx, tgt_idx in hypo['alignment']]) )) return outputs