From b9a42b396f71f494842faf2ad4fb783f5b88b361 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Wed, 8 Jan 2025 08:31:28 -0500 Subject: [PATCH] linting --- tests/e2e/integrations/kd.py | 1 + tests/e2e/integrations/test_liger.py | 2 ++ 2 files changed, 3 insertions(+) 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()