ci fix

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

Author: syhuang22

src/maxdiffusion/tests/test_embeddings_connector_ltx2.py

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