debug

Author: prishajain1

src/maxdiffusion/tests/ltx2_parity_test.py

@@ -603,6 +603,13 @@ def test_import_parity_comparison_split(self):
     exactly (to a high precision) given the same inputs and weights, with rope_type='split'.
     """
     print("\n=== Parity Comparison Test (Split) ===")
+    try:
+      import torch
+    except ImportError:
+      print("Skipping parity test: torch not installed.")
+      return
+
+    import os
     from flax import traverse_util
 
     parity_file = "ltx2_parity_data_split.pt"