From 2b9a808758b59474bb13be633a8c4e1fbff14ed2 Mon Sep 17 00:00:00 2001 From: sanchit-gandhi Date: Sat, 30 Sep 2023 16:39:19 +0100 Subject: [PATCH] revert eq test --- tests/models/falcon/test_modeling_falcon.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/tests/models/falcon/test_modeling_falcon.py b/tests/models/falcon/test_modeling_falcon.py index 072ed40d2bebef..40d8566f66e904 100644 --- a/tests/models/falcon/test_modeling_falcon.py +++ b/tests/models/falcon/test_modeling_falcon.py @@ -590,19 +590,15 @@ def test_auto_config_without_remote_code(self): def test_config_with_remote_code(self): for supported_checkpoint in self.supported_checkpoints: - config1 = FalconConfig.from_pretrained(supported_checkpoint, trust_remote_code=True) - config2 = FalconConfig.from_pretrained(supported_checkpoint) + config = FalconConfig.from_pretrained(supported_checkpoint, trust_remote_code=True) - self.assertEqual(config1.model_type, "falcon") - self.assertEqual(config1.to_dict(), config2.to_dict()) + self.assertEqual(config.model_type, "falcon") def test_auto_config_with_remote_code(self): for supported_checkpoint in self.supported_checkpoints: - config1 = AutoConfig.from_pretrained(supported_checkpoint, trust_remote_code=True) - config2 = FalconConfig.from_pretrained(supported_checkpoint) + config = AutoConfig.from_pretrained(supported_checkpoint, trust_remote_code=True) - self.assertEqual(config1.model_type, "falcon") - self.assertEqual(config1.to_dict(), config2.to_dict()) + self.assertEqual(config.model_type, "falcon") def test_config_with_specific_revision(self): for supported_checkpoint in self.supported_checkpoints: