feat: assert return_dict

modeling_for_glue.py

@@ -55,7 +55,7 @@ class BertForSequenceClassification(BertPreTrainedModel):
         assert inputs_embeds is None
         assert output_attentions is None
         assert output_hidden_states is None
-        assert return_dict is None
+        assert return_dict
         outputs = self.bert(
             input_ids,
             attention_mask=attention_mask,
@@ -147,7 +147,7 @@ class BertForQuestionAnswering(BertPreTrainedModel):
         assert inputs_embeds is None
         assert output_attentions is None
         assert output_hidden_states is None
-        assert return_dict is None
+        assert return_dict
         outputs = self.bert(
             input_ids,
             attention_mask=attention_mask,
@@ -234,7 +234,7 @@ class BertForTokenClassification(BertPreTrainedModel):
         assert inputs_embeds is None
         assert output_attentions is None
         assert output_hidden_states is None
-        assert return_dict is None
+        assert return_dict
         outputs = self.bert(
             input_ids,
             attention_mask=attention_mask,