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_t5.py
+++ b/tests/test_modeling_t5.py
@@ -206,7 +206,7 @@ class T5ModelTest(ModelTesterMixin, unittest.TestCase):
                input_ids=input_ids,
                decoder_input_ids=decoder_input_ids,
                decoder_attention_mask=decoder_attention_mask,
-                lm_labels=lm_labels,
+                labels=lm_labels,
            )
            loss, prediction_scores, _, _ = outputs
            self.parent.assertEqual(len(outputs), 4)