Skip to content

Add embodichain-new-task CLI for task environment scaffolding#279

Draft
yuecideng wants to merge 1 commit into
mainfrom
feat/task-scaffold-cli
Draft

Add embodichain-new-task CLI for task environment scaffolding#279
yuecideng wants to merge 1 commit into
mainfrom
feat/task-scaffold-cli

Conversation

@yuecideng
Copy link
Copy Markdown
Contributor

Description

This PR adds the embodichain-new-task console command and embodichain.toolkits.scaffold package to generate EmbodiChain task environments from Jinja2 templates.

The CLI supports:

  • Targets: in-repo (embodichain/lab/gym/envs/tasks/...) or external extension projects
  • Workflows: expert demo (EmbodiedEnv), RL (RLEnv), and config-only env classes
  • Outputs: task Python module, gym JSON config, pytest stub, optional extension pyproject.toml / README / run_env.py
  • Post-processing: patches tasks/__init__.py for in-repo tasks and can run black on generated files

skills/add-task-env/SKILL.md documents the preferred CLI workflow before manual scaffolding.

Dependencies: jinja2>=3.1.0 (added to pyproject.toml).

Type of change

  • Bug fix (non-breaking change which fixes an issue)
  • Enhancement (non-breaking change which improves an existing functionality)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (existing functionality will not work without user modification)
  • Documentation update

Screenshots

N/A

Checklist

  • I have run the black . command to format the code base.
  • I have made corresponding changes to the documentation
  • I have added tests that prove my fix is effective or that my feature works
  • Dependencies have been updated, if applicable.

Test plan

  • pip install -e . and verify embodichain-new-task --help
  • embodichain-new-task --target inrepo --workflow demo --category special --name scaffold_smoke --gym-id ScaffoldSmoke-v0 --dry-run
  • pytest tests/toolkits/scaffold/test_scaffold.py

🤖 Generated with Claude Code

Made with Cursor

Introduce a Jinja2-based scaffold that generates in-repo and extension
task environments (demo, RL, config-only), with console entry point,
package templates, and unit tests. Document the CLI in add-task-env skill.

Co-authored-by: Cursor <cursoragent@cursor.com>
@yuecideng yuecideng added toolkit Stand along tools collection. gym robot learning env and its related features task A task written in openai gym format for imitation learning or reinforcement learning enhancement New feature or request labels May 23, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request gym robot learning env and its related features task A task written in openai gym format for imitation learning or reinforcement learning toolkit Stand along tools collection.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant