Skip to content

Commit 81f68ff

Browse files
committed
Add Dockerfile for bank_subscription_prediction model
1 parent 5797dcf commit 81f68ff

File tree

1 file changed

+39
-0
lines changed

1 file changed

+39
-0
lines changed
Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,39 @@
1+
# Sandbox base image
2+
FROM zenmldocker/zenml-sandbox:latest
3+
4+
# Install uv from official distroless image
5+
COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
6+
7+
# Set uv environment variables for optimization
8+
ENV UV_SYSTEM_PYTHON=1
9+
ENV UV_COMPILE_BYTECODE=1
10+
11+
# Project metadata
12+
LABEL project_name="bank_subscription_prediction"
13+
LABEL project_version="0.1.0"
14+
15+
# Install dependencies with uv and cache optimization
16+
RUN --mount=type=cache,target=/root/.cache/uv \
17+
uv pip install --system \
18+
"zenml[server]>=0.50.0" \
19+
"notebook" \
20+
"scikit-learn" \
21+
"pyarrow" \
22+
"pandas" \
23+
"xgboost" \
24+
"matplotlib" \
25+
"seaborn" \
26+
"plotly" \
27+
"jupyter"
28+
29+
# Set workspace directory
30+
WORKDIR /workspace
31+
32+
# Clone only the project directory and reorganize
33+
RUN git clone --depth 1 https://github.com/zenml-io/zenml-projects.git /tmp/zenml-projects && \
34+
cp -r /tmp/zenml-projects/bank_subscription_prediction/* /workspace/ && \
35+
rm -rf /tmp/zenml-projects
36+
37+
# VSCode settings
38+
RUN mkdir -p /workspace/.vscode && \
39+
printf '{\n "workbench.colorTheme": "Default Dark Modern"\n}' > /workspace/.vscode/settings.json

0 commit comments

Comments
 (0)