diff --git a/tests/e2e/integrations/kd.py b/tests/e2e/integrations/kd.py index 5f865c87d..147f4fc78 100644 --- a/tests/e2e/integrations/kd.py +++ b/tests/e2e/integrations/kd.py @@ -72,6 +72,7 @@ class TestKnowledgeDistillation: def test_llama_kd(self, temp_dir, kd_min_cfg): cfg = DictDefault(kd_min_cfg) + # pylint: disable=duplicate-code prepare_plugins(cfg) normalize_config(cfg) cli_args = TrainerCliArgs() diff --git a/tests/e2e/integrations/test_liger.py b/tests/e2e/integrations/test_liger.py index 1efe889e4..e95ac640b 100644 --- a/tests/e2e/integrations/test_liger.py +++ b/tests/e2e/integrations/test_liger.py @@ -55,6 +55,7 @@ class LigerIntegrationTestCase: "max_steps": 5, } ) + # pylint: disable=duplicate-code prepare_plugins(cfg) normalize_config(cfg) cli_args = TrainerCliArgs() @@ -100,6 +101,7 @@ class LigerIntegrationTestCase: "max_steps": 5, } ) + # pylint: disable=duplicate-code prepare_plugins(cfg) normalize_config(cfg) cli_args = TrainerCliArgs()