tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

linting

Browse Source
This commit is contained in:
Wing Lian
2025-01-08 08:31:28 -05:00
parent 7e7762f40b
commit a31ec4d9b3
2 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/e2e/integrations/kd.py
View File

@@ -72,6 +72,7 @@ class TestKnowledgeDistillation:
def test_llama_kd(self, temp_dir, kd_min_cfg): def test_llama_kd(self, temp_dir, kd_min_cfg):
cfg = DictDefault(kd_min_cfg) cfg = DictDefault(kd_min_cfg)
# pylint: disable=duplicate-code
prepare_plugins(cfg) prepare_plugins(cfg)
normalize_config(cfg) normalize_config(cfg)
cli_args = TrainerCliArgs() cli_args = TrainerCliArgs()

2
tests/e2e/integrations/test_liger.py
View File

@@ -54,6 +54,7 @@ class LigerIntegrationTestCase:
"max_steps": 5, "max_steps": 5,
} }
) )
# pylint: disable=duplicate-code
prepare_plugins(cfg) prepare_plugins(cfg)
normalize_config(cfg) normalize_config(cfg)
cli_args = TrainerCliArgs() cli_args = TrainerCliArgs()
@@ -99,6 +100,7 @@ class LigerIntegrationTestCase:
"max_steps": 5, "max_steps": 5,
} }
) )
# pylint: disable=duplicate-code
prepare_plugins(cfg) prepare_plugins(cfg)
normalize_config(cfg) normalize_config(cfg)
cli_args = TrainerCliArgs() cli_args = TrainerCliArgs()