From a426b3d002f1928a66d0b1ec0452fb1522d17dfc Mon Sep 17 00:00:00 2001 From: bursteratom Date: Wed, 11 Dec 2024 16:15:44 -0500 Subject: [PATCH] switching test hymba order --- tests/e2e/test_packing_loss.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/tests/e2e/test_packing_loss.py b/tests/e2e/test_packing_loss.py index 43f623ca6..22964be95 100644 --- a/tests/e2e/test_packing_loss.py +++ b/tests/e2e/test_packing_loss.py @@ -70,14 +70,14 @@ def test_loss_packed(self, temp_dir): ) -class TestPackedHymba(unittest.TestCase): +class TestUnpackedHymba(unittest.TestCase): """ - Test case for Packed training of hymba models + Test case for Unpacked training of hymba models """ @require_torch_2_5_1 @with_temp_dir - def test_loss_packed(self, temp_dir): + def test_loss_unpacked(self, temp_dir): # pylint: disable=duplicate-code cfg = DictDefault( { @@ -98,7 +98,7 @@ def test_loss_packed(self, temp_dir): "o_proj", ], "sequence_len": 1024, - "sample_packing": True, + "sample_packing": False, "flash_attention": True, "val_set_size": 0.0, "datasets": [ @@ -133,14 +133,14 @@ def test_loss_packed(self, temp_dir): ) -class TestUnpackedHymba(unittest.TestCase): +class TestPackedHymba(unittest.TestCase): """ - Test case for Unpacked training of hymba models + Test case for Packed training of hymba models """ @require_torch_2_5_1 @with_temp_dir - def test_loss_unpacked(self, temp_dir): + def test_loss_packed(self, temp_dir): # pylint: disable=duplicate-code cfg = DictDefault( { @@ -161,7 +161,7 @@ def test_loss_unpacked(self, temp_dir): "o_proj", ], "sequence_len": 1024, - "sample_packing": False, + "sample_packing": True, "flash_attention": True, "val_set_size": 0.0, "datasets": [