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 ce9299b92..4796c1a43 100644 --- a/tests/e2e/integrations/test_liger.py +++ b/tests/e2e/integrations/test_liger.py @@ -54,6 +54,7 @@ class LigerIntegrationTestCase: "max_steps": 5, } ) + # pylint: disable=duplicate-code prepare_plugins(cfg) normalize_config(cfg) cli_args = TrainerCliArgs() @@ -99,6 +100,7 @@ class LigerIntegrationTestCase: "max_steps": 5, } ) + # pylint: disable=duplicate-code prepare_plugins(cfg) normalize_config(cfg) cli_args = TrainerCliArgs()