Update Dockerfile.ci to use importlib.metadata for retrieving the current version of flashinfer, enhancing compatibility with the latest Python standards.

Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
tzhouam committed 2026-03-03 02:28:24 +00:00
1 parent 65169baa9d
commit 029bb56e3a
1 file changed
+1 -1
+1 -1
View File
@@ -22,7 +22,7 @@ RUN uv pip install --system --upgrade --force-reinstall --prerelease=allow vllm
RUN uv pip install --system --no-cache-dir ".[dev]"
RUN ln -sf /usr/bin/python3 /usr/bin/python
RUN FLASHINFER_VERSION=$(python -c "import flashinfer; print(flashinfer.__version__)") && \
RUN FLASHINFER_VERSION=$(python -c "from importlib.metadata import version; print(version('flashinfer'))") && \
uv pip install --system --upgrade "flashinfer-cubin==${FLASHINFER_VERSION}"
ENTRYPOINT []