From 414e475363cf2858c1810faba9250484b2a6177c Mon Sep 17 00:00:00 2001 From: Yige Xu Date: Tue, 17 Sep 2019 15:34:00 +0800 Subject: [PATCH] fix an warning message error in matching pipe --- fastNLP/io/pipe/matching.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fastNLP/io/pipe/matching.py b/fastNLP/io/pipe/matching.py index 1d22aede..6bf81085 100644 --- a/fastNLP/io/pipe/matching.py +++ b/fastNLP/io/pipe/matching.py @@ -114,7 +114,7 @@ class MatchingBertPipe(Pipe): if ('train' not in name) and (ds.has_field(Const.TARGET))] ) if len(target_vocab._no_create_word) > 0: - warn_msg = f"There are {len(tgt_vocab._no_create_word)} target labels" \ + warn_msg = f"There are {len(target_vocab._no_create_word)} target labels" \ f" in {[name for name in data_bundle.datasets.keys() if 'train' not in name]} " \ f"data set but not in train data set!." warnings.warn(warn_msg) @@ -251,7 +251,7 @@ class MatchingPipe(Pipe): if ('train' not in name) and (ds.has_field(Const.TARGET))] ) if len(target_vocab._no_create_word) > 0: - warn_msg = f"There are {len(tgt_vocab._no_create_word)} target labels" \ + warn_msg = f"There are {len(target_vocab._no_create_word)} target labels" \ f" in {[name for name in data_bundle.datasets.keys() if 'train' not in name]} " \ f"data set but not in train data set!." warnings.warn(warn_msg)