Apply suggestions from code review

Co-authored-by: Michael Goin <[email protected]>
Signed-off-by: csy1204 <[email protected]>

Author: csy1204

docker/Dockerfile

@@ -163,7 +163,8 @@ COPY requirements/lint.txt requirements/lint.txt
 COPY requirements/test.txt requirements/test.txt
 COPY requirements/dev.txt requirements/dev.txt
 # Workaround for #17068
-RUN pip install mamba-ssm==2.2.4           
+RUN --mount=type=cache,target=/root/.cache/uv \
+    uv pip install --system mamba-ssm==2.2.4 --no-build-isolation
 RUN --mount=type=cache,target=/root/.cache/uv \
     uv pip install --system -r requirements/dev.txt
 #################### DEV IMAGE ####################
@@ -268,7 +269,8 @@ ENV UV_HTTP_TIMEOUT=500
 
 # install development dependencies (for testing)
 # Workaround for #17068
-RUN pip install mamba-ssm==2.2.4           
+RUN --mount=type=cache,target=/root/.cache/uv \
+    uv pip install --system mamba-ssm==2.2.4 --no-build-isolation
 RUN --mount=type=cache,target=/root/.cache/uv \
     uv pip install --system -r requirements/dev.txt