Skip to content

Comments

add color sorting task#3

Open
jyizheng wants to merge 1 commit intoVBVR-DataFactory:mainfrom
jyizheng:yizheng/task1
Open

add color sorting task#3
jyizheng wants to merge 1 commit intoVBVR-DataFactory:mainfrom
jyizheng:yizheng/task1

Conversation

@jyizheng
Copy link

@jyizheng jyizheng commented Jan 8, 2026

This pull request adds a new "Color Sorting" visual reasoning task to the codebase, including its generator, configuration, and documentation. The task requires sorting colored blocks into matching containers and includes image and optional video output. The main changes are the implementation of the task generation logic, integration into the codebase, and updates to the documentation and CLI tools.

New Color Sorting Task Implementation:

  • Introduced src/color_sorting_task.py containing the ColorSortingTask, ColorSortingConfig, ColorSortingGenerator, and prompt generation logic for the color sorting task. This includes rendering input/output images, generating animation videos, and supporting multiple color configurations.
  • Added a new CLI script examples/generate_color_sorting.py to generate datasets for the color sorting task, supporting various command-line options for customization.

Codebase Integration:

  • Updated src/__init__.py to import and expose the new color sorting task classes and prompt function, making them available as part of the module's public API.

Documentation:

  • Expanded README.md with a detailed description of the color sorting task, example usage, configuration options, and output structure, making it easy for users to understand and use the new task.

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.

1 participant