Skip to content

Conversation

@XieLipeng0830
Copy link
Collaborator

…nment

OpenJudge Version

[The version of OpenJudge you are working on, e.g. import openjudge; print(openjudge.__version__)]

Description

[Please describe the background, purpose, changes made, and how to test this PR]

Checklist

Please check the following items before code is ready to be reviewed.

  • Code has been formatted with pre-commit run --all-files command
  • All tests are passing
  • Docstrings are in Google style
  • Related documentation has been updated (e.g. links, examples, etc.)
  • Code is ready for review

@gemini-code-assist
Copy link
Contributor

Summary of Changes

Hello @XieLipeng0830, I'm Gemini Code Assist1! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request significantly enhances the OpenJudge project by introducing robust Docker support for both its evaluation and training environments. These additions aim to streamline the setup process, ensure environment consistency, and simplify dependency management for users. The accompanying documentation updates provide clear guidance on leveraging these new containerized solutions, alongside more detailed manual installation steps for complex training setups.

Highlights

  • New Dockerfile for OpenJudge Evaluation: A new Dockerfile has been added to provide a containerized environment for running OpenJudge evaluation tasks, including all necessary system and Python dependencies.
  • New Dockerfile for OpenJudge Training: A dedicated Dockerfile (Dockerfile.train) has been introduced for setting up a comprehensive training environment for Judge models, supporting SFT/RL with PyTorch, CUDA, sglang, vllm, verl, FlashAttention, and other ML packages.
  • Updated Documentation for Training Environment Setup: The docs/building_graders/training_judge_models.md file has been updated to include detailed instructions for setting up the training environment using the new Docker image or a comprehensive manual installation guide for verl and its dependencies.
  • Updated Quickstart Guide with Docker Installation: The docs/get_started/quickstart.md now includes a Docker-based installation option, making it easier for users to get started with OpenJudge using containers.
Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature Command Description
Code Review /gemini review Performs a code review for the current pull request in its current state.
Pull Request Summary /gemini summary Provides a summary of the current pull request in its current state.
Comment @gemini-code-assist Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help /gemini help Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for Github and other Google products, sign up here.

You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension.

Footnotes

  1. Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution.

Copy link
Contributor

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

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 Docker support for both the base OpenJudge environment and the more complex training environment, which is a great addition for improving reproducibility and simplifying setup. The documentation has also been updated accordingly. My review focuses on improving the Dockerfiles for better optimization and reproducibility, and I've also noted a minor typo in the documentation. Overall, these are valuable changes.

Dockerfile Outdated
"loguru>=0.7.3,<0.8.0" \
"json_repair>=0.54.0,<1.0.0" \
"pydantic>=2.11.5,<3.0.0" \
"openai>=1.92.0,<2.0.0" \
Copy link
Contributor

Choose a reason for hiding this comment

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

high

The version requirement for openai here (>=1.92.0,<2.0.0) is inconsistent with what's specified in pyproject.toml and cookbooks/training_judge_model/Dockerfile.train (>=1.85.0,<2.0.0). To ensure consistency across the project and avoid potential dependency conflicts, it's best to align this with the version in pyproject.toml.

    "openai>=1.85.0,<2.0.0" \

"scikit-learn>=1.0.0"

# Install OpenJudge from GitHub
RUN pip install --no-cache-dir git+https://github.com/agentscope-ai/OpenJudge.git
Copy link
Contributor

Choose a reason for hiding this comment

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

high

Installing from a git repository URL without pinning to a specific commit hash or tag can lead to non-reproducible builds, as it will pull the latest commit from the main branch. This might introduce breaking changes unexpectedly. For stable and predictable builds, please pin the installation to a specific commit hash or tag.

For example:

RUN pip install --no-cache-dir git+https://github.com/agentscope-ai/OpenJudge.git@your-commit-hash

# =============================================================================
# 8. OpenJudge
# =============================================================================
RUN pip install --no-cache-dir --no-deps git+https://github.com/agentscope-ai/OpenJudge.git
Copy link
Contributor

Choose a reason for hiding this comment

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

high

Installing from a git repository URL without pinning to a specific commit hash or tag can lead to non-reproducible builds, as it will pull the latest commit from the main branch. This might introduce breaking changes unexpectedly. For stable and predictable builds, please pin the installation to a specific commit hash or tag.

For example:

RUN pip install --no-cache-dir --no-deps git+https://github.com/agentscope-ai/OpenJudge.git@your-commit-hash

Comment on lines +15 to +19
RUN apt-get update && apt-get install -y --no-install-recommends \
git \
curl \
build-essential \
&& rm -rf /var/lib/apt/lists/*
Copy link
Contributor

Choose a reason for hiding this comment

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

medium

To optimize the Docker image size and reduce the number of layers, it's a good practice to chain related RUN commands. You can combine this apt-get installation with the subsequent pip install and pip cache purge commands into a single RUN instruction. This creates a single layer, making the image more compact and potentially speeding up builds.

Comment on lines +29 to +31
RUN pip install --no-cache-dir "sglang[all]==0.5.2" && \
pip install --no-cache-dir torch-memory-saver && \
pip install --no-cache-dir "vllm==0.11.0"
Copy link
Contributor

Choose a reason for hiding this comment

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

medium

These pip install commands can be combined into a single RUN instruction. This improves readability and allows pip to resolve all dependencies in one go.

RUN pip install --no-cache-dir "sglang[all]==0.5.2" torch-memory-saver "vllm==0.11.0"

Comment on lines 47 to 48
conda create -n openjudeg_train python==3.12
conda activate openjudeg_train
Copy link
Contributor

Choose a reason for hiding this comment

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

medium

There appears to be a typo in the conda environment name. It's written as openjudeg_train, but it should likely be openjudge_train for consistency. This will cause an error for users who copy and paste the command.

Suggested change
conda create -n openjudeg_train python==3.12
conda activate openjudeg_train
conda create -n openjudge_train python==3.12
conda activate openjudge_train

@helloml0326 helloml0326 self-requested a review January 27, 2026 07:44
Copy link
Collaborator

@ployts ployts left a comment

Choose a reason for hiding this comment

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

LGTM

@ployts ployts merged commit 393d7cc into agentscope-ai:main Jan 28, 2026
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants