import fix in ltx_2_transformer_test.py

Author: prishajain1

src/maxdiffusion/tests/ltx_2_transformer_test.py

@@ -22,7 +22,7 @@
 from flax import nnx
 from jax.sharding import Mesh
 from flax.linen import partitioning as nn_partitioning
-from .. import pyconfig
+from maxdiffusion import pyconfig
 from ..max_utils import create_device_mesh
 from ..models.ltx2.transformer_ltx2 import (
     LTX2VideoTransformerBlock,