runtime error

Exit code: 1. Reason: Traceback (most recent call last): File "/app/app.py", line 12, in <module> from pipeline import INTEGRATORS, load_pipeline_assets, resolve_integrator, sample_batch File "/app/pipeline.py", line 7, in <module> from flax import nnx, serialization File "/usr/local/lib/python3.13/site-packages/flax/nnx/__init__.py", line 22, in <module> from .bridge import wrappers as wrappers File "/usr/local/lib/python3.13/site-packages/flax/nnx/bridge/__init__.py", line 16, in <module> from .wrappers import functional as functional File "/usr/local/lib/python3.13/site-packages/flax/nnx/bridge/wrappers.py", line 24, in <module> from flax.nnx import graph File "/usr/local/lib/python3.13/site-packages/flax/nnx/graph.py", line 26, in <module> from flax.nnx import filterlib, reprlib, traversals, variablelib File "/usr/local/lib/python3.13/site-packages/flax/nnx/variablelib.py", line 52, in <module> from jax._src.core import mutable_array, MutableArray ImportError: cannot import name 'mutable_array' from 'jax._src.core' (/usr/local/lib/python3.13/site-packages/jax/_src/core.py)

Container logs: