SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix test_configuration_tie in FlaxEncoderDecoderModelTest (#14076)

Browse Source 
* 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 <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2021-11-02 11:02:41 +01:00
committed by GitHub
parent a767276fdd
commit 4a394cf53f
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_modeling_flax_encoder_decoder.py
View File

@@ -357,8 +357,11 @@ class FlaxEncoderDecoderModelTest(unittest.TestCase):
return config return config
def _check_configuration_tie(self, model): 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 @slow
def test_configuration_tie(self): def test_configuration_tie(self):