Skip to content

Commit 7991726

Browse files
committed
Add installation for HFResourceScanner if enabled in Dockerfile
Signed-off-by: Angel Luu <angel.luu@us.ibm.com>
1 parent 37f61bb commit 7991726

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

build/Dockerfile

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ ARG WHEEL_VERSION=""
2222
## Enable Aimstack if requested via ENABLE_AIM set to "true"
2323
ARG ENABLE_AIM=false
2424
ARG ENABLE_FMS_ACCELERATION=true
25+
ARG ENABLE_SCANNER=false
2526

2627
## Base Layer ##################################################################
2728
FROM registry.access.redhat.com/ubi9/ubi:${BASE_UBI_IMAGE_TAG} AS base
@@ -110,6 +111,7 @@ ARG USER
110111
ARG USER_UID
111112
ARG ENABLE_FMS_ACCELERATION
112113
ARG ENABLE_AIM
114+
ARG ENABLE_SCANNER
113115

114116
RUN dnf install -y git && \
115117
# perl-Net-SSLeay.x86_64 and server_key.pem are installed with git as dependencies
@@ -151,6 +153,10 @@ RUN if [[ "${ENABLE_AIM}" == "true" ]]; then \
151153
python -m pip install --user "$(head bdist_name)[aim]"; \
152154
fi
153155

156+
RUN if [[ "${ENABLE_SCANNER}}" == "true" ]]; then \
157+
python -m pip install --user "$(head bdist_name)[scanner-dev]"; \
158+
fi
159+
154160
# Clean up the wheel module. It's only needed by flash-attn install
155161
RUN python -m pip uninstall wheel build -y && \
156162
# Cleanup the bdist whl file

0 commit comments

Comments
 (0)