Purge unused ModelTester code (#37085)
* Purge correctly this time * Remove more methods from recent PRs * make fixup
This commit is contained in:
@@ -180,15 +180,6 @@ class CTRLModelTester:
|
||||
|
||||
return config, inputs_dict
|
||||
|
||||
def create_and_check_ctrl_for_sequence_classification(self, config, input_ids, head_mask, token_type_ids, *args):
|
||||
config.num_labels = self.num_labels
|
||||
model = CTRLForSequenceClassification(config)
|
||||
model.to(torch_device)
|
||||
model.eval()
|
||||
sequence_labels = ids_tensor([self.batch_size], self.type_sequence_label_size)
|
||||
result = model(input_ids, token_type_ids=token_type_ids, labels=sequence_labels)
|
||||
self.parent.assertEqual(result.logits.shape, (self.batch_size, self.num_labels))
|
||||
|
||||
|
||||
@require_torch
|
||||
class CTRLModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin, unittest.TestCase):
|
||||
|
||||
Reference in New Issue
Block a user