linting

2025-01-08 08:31:28 -05:00
parent f7963083b8
commit 58d9896777
2 changed files with 3 additions and 0 deletions
--- 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()
--- 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()