Skip to content

GitLab

Unverified Commit 73ad2588 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix tests without any real effect (#13406) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 76c4d8bf
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/test_modeling_encoder_decoder.py
View file @ 73ad2588
...@@ -200,7 +200,7 @@ class EncoderDecoderMixin: ...@@ -200,7 +200,7 @@ class EncoderDecoderMixin:
with tempfile.TemporaryDirectory() as tmpdirname: with tempfile.TemporaryDirectory() as tmpdirname:
enc_dec_model.save_pretrained(tmpdirname) enc_dec_model.save_pretrained(tmpdirname)
EncoderDecoderModel.from_pretrained(tmpdirname) enc_dec_model = EncoderDecoderModel.from_pretrained(tmpdirname)
after_outputs = enc_dec_model( after_outputs = enc_dec_model(
input_ids=input_ids, input_ids=input_ids,
...@@ -241,7 +241,7 @@ class EncoderDecoderMixin: ...@@ -241,7 +241,7 @@ class EncoderDecoderMixin:
with tempfile.TemporaryDirectory() as encoder_tmp_dirname, tempfile.TemporaryDirectory() as decoder_tmp_dirname: 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.encoder.save_pretrained(encoder_tmp_dirname)
enc_dec_model.decoder.save_pretrained(decoder_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, encoder_pretrained_model_name_or_path=encoder_tmp_dirname,
decoder_pretrained_model_name_or_path=decoder_tmp_dirname, decoder_pretrained_model_name_or_path=decoder_tmp_dirname,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment