debug statement

Author: prishajain1

src/maxdiffusion/tests/ltx2_parity_test.py

@@ -601,6 +601,7 @@ def test_import_parity_comparison_split(self):
     Verifies that the LTX2VideoTransformer3DModel output matches the PyTorch implementation output
     exactly (to a high precision) given the same inputs and weights, with rope_type='split'.
     """
+    print("\n=== Parity Comparison Test (Split) ===")
     from flax import traverse_util
 
     parity_file = "ltx2_parity_data_split.pt"