From bcb1205e39ddc32916f8578dfddeee98ed3d183f Mon Sep 17 00:00:00 2001 From: sunny Date: Tue, 5 Nov 2024 12:30:45 -0500 Subject: [PATCH] test --- tests/e2e/test_dpo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/e2e/test_dpo.py b/tests/e2e/test_dpo.py index 156949fd8..db1e0f117 100644 --- a/tests/e2e/test_dpo.py +++ b/tests/e2e/test_dpo.py @@ -116,7 +116,7 @@ class TestDPOLlamaLora(unittest.TestCase): assert (Path(temp_dir) / "checkpoint-20/adapter_model.safetensors").exists() @with_temp_dir - def test_dpo_nll_use_weighting(self, temp_dir): + def test_dpo_use_weighting(self, temp_dir): # pylint: disable=duplicate-code cfg = DictDefault( {