ci fix

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

Author: syhuang22

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

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