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_openai.py
+++ b/tests/test_modeling_openai.py
@@ -169,7 +169,7 @@ class OpenAIGPTModelTest(ModelTesterMixin, unittest.TestCase):
            model.to(torch_device)
            model.eval()

-            loss, lm_logits, mc_logits = model(input_ids, token_type_ids=token_type_ids, lm_labels=input_ids)
+            loss, lm_logits, mc_logits = model(input_ids, token_type_ids=token_type_ids, labels=input_ids)

            result = {"loss": loss, "lm_logits": lm_logits}