fix incorrect logging in viterbi decode (#2982)

allenai · Jun 20, 2019 · 03aa838 · 03aa838
1 parent 51b74b1
commit 03aa838
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/allennlp/nn/util.py b/allennlp/nn/util.py
@@ -509,7 +509,7 @@ def viterbi_decode(tag_sequence: torch.Tensor,
         observation = tag_observations[timestep]
         # Warn the user if they have passed
         # invalid/extremely unlikely evidence.
-        if tag_observations[timestep - 1] != -1:
+        if tag_observations[timestep - 1] != -1 and observation != -1:
             if transition_matrix[tag_observations[timestep - 1], observation] < -10000:
                 logger.warning("The pairwise potential between tags you have passed as "
                                "observations is extremely unlikely. Double check your evidence "