Unwrap masks to tensors as well in AUC metric (#2813)

Varal7 · DeNeutoy · commit 7c4311b11de7 · 2019-05-10T12:17:23.000-07:00
diff --git a/allennlp/training/metrics/auc.py b/allennlp/training/metrics/auc.py
@@ -37,7 +37,7 @@ def __call__(self,
             A one-dimensional label tensor of shape (batch_size).
         """
 
-        predictions, gold_labels = self.unwrap_to_tensors(predictions, gold_labels)
+        predictions, gold_labels, mask = self.unwrap_to_tensors(predictions, gold_labels, mask)
 
         # Sanity checks.
         if gold_labels.dim() != 1:
