Skip to content

Commit bc220af

Browse files
authored
♾️ [CI] Remove test_raise_error_not_causallm (huggingface#3265)
1 parent f427303 commit bc220af

File tree

1 file changed

+1
-9
lines changed

1 file changed

+1
-9
lines changed

tests/test_modeling_value_head.py

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818

1919
import torch
2020
from parameterized import parameterized
21-
from transformers import AutoModel, AutoModelForCausalLM, AutoModelForSeq2SeqLM, GenerationConfig
21+
from transformers import AutoModelForCausalLM, AutoModelForSeq2SeqLM, GenerationConfig
2222

2323
from trl import AutoModelForCausalLMWithValueHead, AutoModelForSeq2SeqLMWithValueHead, create_reference_model
2424

@@ -381,14 +381,6 @@ def test_generate(self, model_name):
381381
# Just check if the generation works
382382
_ = model.generate(input_ids, decoder_input_ids=decoder_input_ids, generation_config=generation_config)
383383

384-
def test_raise_error_not_causallm(self):
385-
# Test with a model without a LM head
386-
model_id = "trl-internal-testing/tiny-T5ForConditionalGeneration"
387-
# This should raise a ValueError
388-
with self.assertRaises(ValueError):
389-
pretrained_model = AutoModel.from_pretrained(model_id)
390-
_ = self.trl_model_class.from_pretrained(pretrained_model)
391-
392384
@unittest.skip("This test needs to be run manually due to HF token issue.")
393385
def test_push_to_hub(self):
394386
for model_name in self.all_model_names:

0 commit comments

Comments
 (0)