Skip to content

Commit c01e711

Browse files
Stop inheriting tests! (#42192)
* Stop inheriting tests! * Just use a del instead * fixup
1 parent 082e3ff commit c01e711

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

tests/models/cohere2/test_modeling_cohere2.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -269,3 +269,6 @@ def test_generation_beyond_sliding_window(self, attn_implementation: str):
269269
output_text = tokenizer.batch_decode(out)
270270

271271
self.assertEqual(output_text, EXPECTED_COMPLETIONS)
272+
273+
274+
del CohereModelTest, CohereModelTester # So the parent tests don't run in this file too

tests/models/granitemoehybrid/test_modeling_granitemoehybrid.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -178,3 +178,6 @@ def test_model_generation(self):
178178
text = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
179179

180180
self.assertEqual(EXPECTED_TEXT_COMPLETION, text)
181+
182+
183+
del BambaModelTest, BambaModelTester # So the parent tests don't run in this file too

0 commit comments

Comments
 (0)