pipeline test unittest fix

Author: prishajain1

src/maxdiffusion/tests/ltx2/test_pipeline_ltx2.py

@@ -118,9 +118,13 @@ def test_check_inputs(self):
     with self.assertRaises(ValueError):
       pipeline.check_inputs(prompt="test", height=64, width=63)
 
+  @patch("maxdiffusion.pipelines.ltx2.ltx2_pipeline.get_tpu_type")
   @patch("maxdiffusion.pipelines.ltx2.ltx2_pipeline.LTX2Pipeline._get_gemma_prompt_embeds")
-  def test_encode_prompt(self, list_embed_mock):
+  def test_encode_prompt(self, list_embed_mock, mock_get_tpu_type):
     """Test conditional encoding of positive and negative prompts."""
+    from maxdiffusion.tpu_utils import TpuType
+    mock_get_tpu_type.return_value = TpuType.TPU_7X
+
     pipeline = LTX2Pipeline(
         scheduler=MagicMock(),
         vae=MagicMock(),