Skip to content

Commit c43ebd2

Browse files
authored
build: enable vllm runtime container as default container for ci pipelines (#1451)
1 parent a298ce9 commit c43ebd2

File tree

2 files changed

+12
-3
lines changed

2 files changed

+12
-3
lines changed

container/Dockerfile.vllm

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -483,6 +483,7 @@ COPY --from=ci_minimum /opt/dynamo/bindings /opt/dynamo/bindings
483483
# Copy nats and etcd from base image
484484
COPY --from=base /usr/bin/nats-server /usr/bin/nats-server
485485
COPY --from=base /usr/local/bin/etcd/ /usr/local/bin/etcd/
486+
ENV PATH=/usr/local/bin/etcd/:$PATH
486487

487488
# Copy UCX from base image as plugin for NIXL
488489
# Copy NIXL source from base image (required for NIXL plugins)
@@ -501,6 +502,15 @@ RUN uv venv $VIRTUAL_ENV --python 3.12 && \
501502
RUN --mount=type=bind,source=./container/deps/requirements.txt,target=/tmp/requirements.txt \
502503
uv pip install --requirement /tmp/requirements.txt
503504

505+
# Install test dependencies
506+
#TODO: Remove this once we have a functional ci_minimum image built on top of the runtime image
507+
RUN --mount=type=bind,source=./container/deps/requirements.test.txt,target=/tmp/requirements.txt \
508+
uv pip install --requirement /tmp/requirements.txt
509+
510+
#TODO: Remove this once we have a functional ci_minimum image built on top of the runtime image
511+
COPY . /workspace
512+
RUN uv pip install /workspace/benchmarks
513+
504514
# Install the wheels and symlink executables to /usr/local/bin so dynamo components can use them
505515
# Dynamo components currently do not have the VIRTUAL_ENV in their PATH, so we need to symlink the executables
506516
#Copy NIXL and Dynamo wheels into wheelhouse
@@ -519,8 +529,6 @@ RUN --mount=type=bind,source=./container/launch_message.txt,target=/workspace/la
519529
sed '/^#\s/d' /workspace/launch_message.txt > ~/.launch_screen && \
520530
echo "cat ~/.launch_screen" >> ~/.bashrc
521531

522-
# Copy examples
523-
COPY ./examples examples/
524532

525-
ENTRYPOINT [ "/usr/bin/bash" ]
533+
ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]
526534
CMD []

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,7 @@ filterwarnings = [
149149
"ignore:.*pkg_resources.*:DeprecationWarning",
150150
"ignore:.*pkg_resources.*:UserWarning",
151151
"ignore:.*multipart.*:PendingDeprecationWarning",
152+
"ignore:.*PyType_Spec.*custom tp_new.*:DeprecationWarning", # Ignore protobuf deprecation warning
152153
]
153154

154155

0 commit comments

Comments
 (0)