From 73ad258806e26eddb354c830d35f06f771f4f2e4 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 6 Sep 2021 14:51:45 +0200 Subject: [PATCH] Fix tests without any real effect (#13406) Co-authored-by: ydshieh --- tests/test_modeling_encoder_decoder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_modeling_encoder_decoder.py b/tests/test_modeling_encoder_decoder.py index 71076d4f21..5f7dce94df 100644 --- a/tests/test_modeling_encoder_decoder.py +++ b/tests/test_modeling_encoder_decoder.py @@ -200,7 +200,7 @@ class EncoderDecoderMixin: with tempfile.TemporaryDirectory() as tmpdirname: enc_dec_model.save_pretrained(tmpdirname) - EncoderDecoderModel.from_pretrained(tmpdirname) + enc_dec_model = EncoderDecoderModel.from_pretrained(tmpdirname) after_outputs = enc_dec_model( input_ids=input_ids, @@ -241,7 +241,7 @@ class EncoderDecoderMixin: with tempfile.TemporaryDirectory() as encoder_tmp_dirname, tempfile.TemporaryDirectory() as decoder_tmp_dirname: enc_dec_model.encoder.save_pretrained(encoder_tmp_dirname) enc_dec_model.decoder.save_pretrained(decoder_tmp_dirname) - EncoderDecoderModel.from_encoder_decoder_pretrained( + enc_dec_model = EncoderDecoderModel.from_encoder_decoder_pretrained( encoder_pretrained_model_name_or_path=encoder_tmp_dirname, decoder_pretrained_model_name_or_path=decoder_tmp_dirname, )