Model output test (#6155)

* Use return_dict=True in all tests * Formatting
2020-07-31 09:44:37 -04:00
parent 86caab1e0b
commit d951c14ae4
26 changed files with 320 additions and 765 deletions
--- a/tests/test_modeling_common.py
+++ b/tests/test_modeling_common.py
@@ -74,7 +74,6 @@ class ModelTesterMixin:

    def test_save_load(self):
        config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
-        config.return_dict = True

        for model_class in self.all_model_classes:
            model = model_class(config)