ci fix

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

Author: syhuang22

src/maxdiffusion/tests/test_feature_extractor_ltx2.py

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