ci fix

Signed-off-by: James Huang <syhuang1201@gmail.com>

Author: syhuang22

src/maxdiffusion/models/ltx2/text_encoders/text_encoders_ltx2.py

@@ -14,7 +14,7 @@
 limitations under the License.
 """
 
-from typing import Optional, Tuple, Union, List
+from typing import Tuple, Union, List
 import jax
 import jax.numpy as jnp
 from flax import nnx

src/maxdiffusion/tests/test_text_encoders_ltx2.py

@@ -15,7 +15,6 @@
 """
 
 import unittest
-import jax
 import jax.numpy as jnp
 import numpy as np
 from flax import nnx