fix

Author: prishajain1

src/maxdiffusion/tests/ltx2_pipeline_test.py

@@ -91,6 +91,7 @@ def setUp(self):
 
         try:
             from maxdiffusion.pipelines.ltx2.ltx2_pipeline import MaxTextGemma3FeatureExtractor
+            from MaxText import common_types
             # Partial mock for config if needed
             class MockConfig:
                 vocab_size = 32000