Use labels to remove deprecation warnings (#4807)

2020-06-05 16:41:46 -04:00
parent 5c0cfc2cf0
commit f1fe18465d
10 changed files with 17 additions and 17 deletions
--- a/tests/test_modeling_electra.py
+++ b/tests/test_modeling_electra.py
@@ -180,7 +180,7 @@ class ElectraModelTest(ModelTesterMixin, unittest.TestCase):
            model.to(torch_device)
            model.eval()
            loss, prediction_scores = model(
-                input_ids, attention_mask=input_mask, token_type_ids=token_type_ids, masked_lm_labels=token_labels
+                input_ids, attention_mask=input_mask, token_type_ids=token_type_ids, labels=token_labels
            )
            result = {
                "loss": loss,