Unverified Commit 4a394cf5 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

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: default avatarydshieh <ydshieh@users.noreply.github.com>
parent a767276f
...@@ -357,8 +357,11 @@ class FlaxEncoderDecoderModelTest(unittest.TestCase): ...@@ -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):
......
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