Skip to content

Commit 8ecafd5

Browse files
authored
Pre-install jq and runner deps in helion Docker image (#463)
Bake jq and the Python packages that github-runner.py actually imports into the image so the workflow doesn't re-download them on every run (~30s saving).
1 parent 1f8f93e commit 8ecafd5

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

docker/helion.Dockerfile

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,18 @@ RUN sudo uv pip install --system flash-linear-attention
6868
# --no-build-isolation: use system torch (cu130) instead of build env pulling cu128 from PyPI
6969
RUN sudo uv pip install --system --no-build-isolation causal-conv1d
7070

71+
# jq for payload extraction in the workflow
72+
RUN sudo apt-get update && sudo apt-get install -y --no-install-recommends jq \
73+
&& sudo rm -rf /var/lib/apt/lists/*
74+
75+
# Kernelbot runner dependencies (only what github-runner.py / run_eval.py actually imports)
76+
RUN sudo uv pip install --system \
77+
PyGithub \
78+
aiohttp \
79+
python-dotenv \
80+
requests \
81+
PyYAML
82+
7183
# # tinygrad
7284
# RUN sudo uv pip install --system tinygrad~=0.10
7385

0 commit comments

Comments
 (0)