From 4a394cf53f05e73ab9bbb4b179a40236a5ffe45a Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Tue, 2 Nov 2021 11:02:41 +0100 Subject: [PATCH] Fix test_configuration_tie in FlaxEncoderDecoderModelTest (#14076) * check test_configuration_tie * Fix test_configuration_tie * make test slow again * Remove property and use model.module.bind * revert to slow test Co-authored-by: ydshieh --- tests/test_modeling_flax_encoder_decoder.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/test_modeling_flax_encoder_decoder.py b/tests/test_modeling_flax_encoder_decoder.py index 3d0311cf85..b23d8ffcf1 100644 --- a/tests/test_modeling_flax_encoder_decoder.py +++ b/tests/test_modeling_flax_encoder_decoder.py @@ -357,8 +357,11 @@ class FlaxEncoderDecoderModelTest(unittest.TestCase): return config def _check_configuration_tie(self, model): - assert id(model.decoder.config) == id(model.config.decoder) - assert id(model.encoder.config) == id(model.config.encoder) + + module = model.module.bind(model.params) + + assert id(module.decoder.config) == id(model.config.decoder) + assert id(module.encoder.config) == id(model.config.encoder) @slow def test_configuration_tie(self):