switching test hymba order

axolotl-ai-cloud · Dec 20, 2024 · 26cd287 · 26cd287
1 parent cce7007
commit 26cd287
Showing 1 changed file with 8 additions and 8 deletions.
diff --git 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": [