linting

2025-01-08 08:31:28 -05:00
parent 7e7762f40b
commit a31ec4d9b3
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
@@ -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()