Skip to content

[sub PR for merge] Add dockerfile and build script#339

Merged
Keithwwa merged 10 commits into
AISBench:masterfrom
SJTUyh:docker_0
Jun 11, 2026
Merged

[sub PR for merge] Add dockerfile and build script#339
Keithwwa merged 10 commits into
AISBench:masterfrom
SJTUyh:docker_0

Conversation

@SJTUyh

@SJTUyh SJTUyh commented Jun 11, 2026

Copy link
Copy Markdown
Collaborator

Thanks for your contribution; we appreciate it a lot. The following instructions will make your pull request healthier and help you get feedback more easily. If you do not understand some items, don't worry, just make the pull request and seek help from maintainers.
感谢您的贡献,我们非常重视。以下说明将使您的拉取请求更健康,更易于获得反馈。如果您不理解某些项目,请不要担心,只需提交拉取请求并从维护人员那里寻求帮助即可。

PR Type / PR类型

  • Feature(功能新增)
  • Bugfix(Bug 修复)
  • Docs(文档更新)
  • CI/CD(持续集成/持续部署)
  • Refactor(代码重构)
  • Perf(性能优化)
  • Dependency(依赖项更新)
  • Test-Cases(测试用例更新)
  • Other(其他)

Related Issue | 关联 Issue
Fixes #(issue ID / issue 编号) / Relates to #(issue ID / issue 编号)

🔍 Motivation / 变更动机

Please describe the motivation of this PR and the goal you want to achieve through this PR.
请描述您的拉取请求的动机和您希望通过此拉取请求实现的目标。

#332

📝 Modification / 修改内容

Please briefly describe what modification is made in this PR.
请简要描述此拉取请求中进行的修改。

📐 Associated Test Results / 关联测试结果

Please provide links to the related test results, such as CI pipelines, test reports, etc.
请提供相关测试结果的链接,例如 CI 管道、测试报告等。

⚠️ BC-breaking (Optional) / 向后不兼容变更(可选)

Does the modification introduce changes that break the backward compatibility of the downstream repositories? If so, please describe how it breaks the compatibility and how the downstream projects should modify their code to keep compatibility with this PR.
是否引入了会破坏下游存储库向后兼容性的更改?如果是,请描述它如何破坏兼容性,以及下游项目应该如何修改其代码以保持与此 PR 的兼容性。

⚠️ Performance degradation (Optional) / 性能下降(可选)

If the modification introduces performance degradation, please describe the impact of the performance degradation and the expected performance improvement.
如果引入了性能下降,请描述性能下降的影响和预期的性能改进。

🌟 Use cases (Optional) / 使用案例(可选)

If this PR introduces a new feature, it is better to list some use cases here and update the documentation.
如果此拉取请求引入了新功能,最好在此处列出一些用例并更新文档。

✅ Checklist / 检查列表

Before PR:

  • Pre-commit or other linting tools are used to fix the potential lint issues. / 使用预提交或其他 linting 工具来修复潜在的 lint 问题。
  • Bug fixes are fully covered by unit tests, the case that causes the bug should be added in the unit tests. / 修复的 Bug 已完全由单元测试覆盖,导致 Bug 的情况应在单元测试中添加。
  • The modification is covered by complete unit tests. If not, please add more unit tests to ensure the correctness. / 此拉取请求中的修改已完全由单元测试覆盖。如果不是,请添加更多单元测试以确保正确性。
  • All relevant documentation (API docs, docstrings, example tutorials) has been updated to reflect these changes. / 所有相关文档(API 文档、文档字符串、示例教程)已更新以反映这些更改。

After PR:

  • If the modification has potential influence on downstream or other related projects, this PR should be tested with those projects. / 如果此拉取请求对下游或其他相关项目有潜在影响,应在那些项目中测试此 PR。
  • CLA has been signed and all committers have signed the CLA in this PR. / CLA 已签署,且本 PR 中的所有提交者均已签署 CLA。

👥 Collaboration Info / 协作信息

  • Suggested Reviewers / 建议审核人: @xxx
  • Relevant Module Owners / 相关模块负责人: @xxx
  • Other Collaboration Notes / 其他协作说明:

🌟 Useful CI Command / 实用的CI命令

Command / 命令 Introduction / 介绍
/gemini review Performs a code review for the current pull request in its current state by Gemini. / 对当前拉取请求在当前状态下由 Gemini 执行代码审核。
/gemini summary Provides a summary of the current pull request in its current state by Gemini. / 对当前拉取请求在当前状态下由 Gemini 提供摘要。
/gemini help Displays a list of available commands of Gemini. / 显示 Gemini 可用命令的列表。
/readthedocs build Triggers a build of the documentation for the current pull request in its current state by Read the Docs. / 触发当前拉取请求在当前状态下由 Read the Docs 构建文档。

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces a Docker image build script (docker/build_image.sh) and several multi-stage Dockerfiles for Ubuntu and openEuler environments to support different Python versions. Additionally, it updates a SWE-bench configuration file. Feedback on these changes highlights a critical syntax error due to unresolved git conflict markers in the SWE-bench configuration. In the Docker build script, improvements are recommended to prevent word splitting by double-quoting variables, handle potential directory navigation failures, and correct an invalid usage example. Finally, the Dockerfiles should be refactored to address redundant package installations and fragile file-copying practices across build stages, either by simplifying them to single-stage builds or properly utilizing Python virtual environments.

Important

The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.

Comment on lines +27 to +31
<<<<<<< HEAD
path="", # necessary for mini datasets, get dataset from https://modelers.cn/datasets/AISBench/SWE-Bench_Multilingual_mini
=======
path="", # Set local path to the mini dataset. Download from https://modelers.cn/datasets/AISBench/SWE-Bench_Multilingual_mini
>>>>>>> master_center

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

critical

Unresolved git conflict markers (<<<<<<< HEAD, =======, >>>>>>> master_center) are present in this file. This will cause a SyntaxError when the Python interpreter attempts to load or run this configuration file. Please resolve the conflict and remove the markers.

        path="",  # Set local path to the mini dataset. Download from https://modelers.cn/datasets/AISBench/SWE-Bench_Multilingual_mini

Comment thread docker/build_image.sh
echo ""
echo "示例:"
echo " $0 --tag v3.1-20260522-master"
echo " $0 --tag v3.1-20260522-master --os ubuntu22.04 --py-version py311"

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

The usage example suggests running with --os ubuntu22.04 --py-version py311. However, there is no corresponding Dockerfile.py311.ubuntu22.04 in the repository (only py310.ubuntu22.04 and py312.ubuntu24.04 are provided for Ubuntu). Running this example command will result in a 'Dockerfile does not exist' error. Please update the example to use a valid combination, such as --os ubuntu24.04 --py-version py312 or --os openeuler24.03 --py-version py311.

Suggested change
echo " $0 --tag v3.1-20260522-master --os ubuntu22.04 --py-version py311"
echo " $0 --tag v3.1-20260522-master --os ubuntu24.04 --py-version py312"

Comment thread docker/build_image.sh
Comment on lines +112 to +113
if docker images -q ${image_name} > /dev/null 2>&1; then
docker rmi -f ${image_name}

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

Double-quote ${image_name} to prevent word splitting if the image name contains spaces or special characters.

Suggested change
if docker images -q ${image_name} > /dev/null 2>&1; then
docker rmi -f ${image_name}
if docker images -q "${image_name}" > /dev/null 2>&1; then
docker rmi -f "${image_name}"

Comment thread docker/build_image.sh
Comment on lines +125 to +130
docker build \
--network host \
--build-arg GIT_TAG=${TAG} \
-f ${dockerfile_path} \
-t ${image_name} \
$(dirname "$0")

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

Variables and command substitutions passed to docker build should be double-quoted to prevent word splitting if paths or tags contain spaces.

Suggested change
docker build \
--network host \
--build-arg GIT_TAG=${TAG} \
-f ${dockerfile_path} \
-t ${image_name} \
$(dirname "$0")
docker build \
--network host \
--build-arg GIT_TAG="${TAG}" \
-f "${dockerfile_path}" \
-t "${image_name}" \
"$(dirname "$0")"

Comment thread docker/build_image.sh
Comment on lines +134 to +140
docker build \
--no-cache \
--network host \
--build-arg GIT_TAG=${TAG} \
-f ${dockerfile_path} \
-t ${image_name} \
$(dirname "$0")

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

Variables and command substitutions passed to docker build should be double-quoted to prevent word splitting if paths or tags contain spaces.

Suggested change
docker build \
--no-cache \
--network host \
--build-arg GIT_TAG=${TAG} \
-f ${dockerfile_path} \
-t ${image_name} \
$(dirname "$0")
docker build \
--no-cache \
--network host \
--build-arg GIT_TAG="${TAG}" \
-f "${dockerfile_path}" \
-t "${image_name}" \
"$(dirname "$0")"

Comment thread docker/build_image.sh
Comment on lines +213 to +214
cd ${obsutils_path}
./obsutil cp ${offline_pkg_full_path} obs://aisbench/images/benchmark/github/${offline_pkg_name} -f

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

If the cd command fails, the script will continue executing ./obsutil in the current working directory, which can lead to unexpected behavior or failures. Additionally, variables like ${obsutils_path}, ${offline_pkg_full_path}, and ${offline_pkg_name} should be double-quoted to prevent word splitting and globbing issues if they contain spaces or special characters.

Suggested change
cd ${obsutils_path}
./obsutil cp ${offline_pkg_full_path} obs://aisbench/images/benchmark/github/${offline_pkg_name} -f
cd "${obsutils_path}" || exit 1
./obsutil cp "${offline_pkg_full_path}" "obs://aisbench/images/benchmark/github/${offline_pkg_name}" -f

Comment on lines +108 to +112
RUN ln -s /usr/local/bin/python3.10 /usr/local/bin/python3 && \
ln -s /usr/local/bin/pip3.10 /usr/local/bin/pip3

COPY --from=builder /usr/local/lib/python3.10 /usr/local/lib/python3.10
COPY --from=builder /usr/local/bin /usr/local/bin

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

Creating symlinks in /usr/local/bin before copying /usr/local/bin from the builder stage is redundant and risky. The COPY instruction will overwrite the directory contents, and the Python make install in the builder stage already creates the necessary python3 and pip3 symlinks. These lines should be removed, or if symlinks are indeed needed, they should be created after the COPY step.

COPY --from=builder /usr/local/lib/python3.10 /usr/local/lib/python3.10
COPY --from=builder /usr/local/bin /usr/local/bin

Comment on lines +145 to +149
COPY --from=builder /usr/local/lib/python3.10 /usr/local/lib/python3.10
COPY --from=builder /usr/local/bin /usr/local/bin
COPY --from=builder /usr/lib/python3/dist-packages /usr/lib/python3/dist-packages
COPY --from=builder /benchmark /benchmark
COPY --from=builder /usr/share/nltk_data /usr/share/nltk_data

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

The current multi-stage build pattern is highly inefficient and defeats the purpose of multi-stage builds.

  1. Redundancy: You are installing the exact same heavy development dependencies (like python3.10-dev, python3-pip, git, curl, vim) in both the builder stage and the final stage.
  2. Fragility: Copying system directories like /usr/local/lib/python3.10, /usr/local/bin, and /usr/lib/python3/dist-packages from the builder stage is extremely fragile. It can easily overwrite or conflict with files installed by apt in the final stage, leading to broken dependencies or runtime issues.

Recommended Solutions:

  • Option A (Single-stage build): Since the final stage installs almost the same packages as the builder stage, you can simplify this into a single-stage Dockerfile. This reduces complexity and avoids fragile directory copying.
  • Option B (Proper Multi-stage with Virtual Environment): Use a Python virtual environment (e.g., in /opt/venv) in the builder stage to install all pip dependencies. In the final stage, install only the runtime dependencies (e.g., python3.10, python3.10-distutils without -dev or pip), copy /opt/venv from the builder, and prepend /opt/venv/bin to the PATH. This keeps the final image clean and significantly smaller.

Comment on lines +143 to +147
COPY --from=builder /usr/local/lib/python3.12 /usr/local/lib/python3.12
COPY --from=builder /usr/local/bin /usr/local/bin
COPY --from=builder /usr/lib/python3/dist-packages /usr/lib/python3/dist-packages
COPY --from=builder /benchmark /benchmark
COPY --from=builder /usr/share/nltk_data /usr/share/nltk_data

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

Similar to the Python 3.10 Dockerfile, this multi-stage build installs the same development dependencies (like python3.12-dev, python3-pip) in both stages and manually copies system python directories (/usr/local/lib/python3.12, /usr/lib/python3/dist-packages). This is fragile and redundant. Please consider either simplifying this to a single-stage build or using a Python virtual environment (venv) to cleanly copy only the installed packages to the final stage without copying system directories.

Comment on lines +100 to +105
COPY --from=builder /usr/share/nltk_data /usr/share/nltk_data
COPY --from=builder /usr/local/bin /usr/local/bin
COPY --from=builder /usr/local/lib64/python3.11/site-packages /usr/local/lib64/python3.11/site-packages
COPY --from=builder /usr/local/lib/python3.11/site-packages /usr/local/lib/python3.11/site-packages
COPY --from=builder /usr/lib64/python3.11/site-packages /usr/lib64/python3.11/site-packages
COPY --from=builder /usr/lib/python3.11/site-packages /usr/lib/python3.11/site-packages

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

This multi-stage build manually copies multiple site-packages directories from the builder stage. This is extremely fragile and prone to breaking Python's package resolution or overwriting system packages. It is highly recommended to use a Python virtual environment (venv) in the builder stage and copy only the virtual environment directory to the final stage, or simplify this to a single-stage build.

@Keithwwa Keithwwa merged commit e67dfeb into AISBench:master Jun 11, 2026
6 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants