From a31ec4d9b301f5dcd16b0ce062424cff7226051e Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
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 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()