Skip to content

feat(inference): show the task field for any policy not known to ignore it - #135

Merged
nicolas-rabault merged 1 commit into
huggingface:mainfrom
ravediamond:inference-task-field
Oct 29, 2026
Merged

nicolas-rabault merged 1 commit into
huggingface:mainfrom
ravediamond:inference-task-field

Conversation

@ravediamond

Copy link
Copy Markdown
Contributor

Follow-up to the discussion on #114.

The inference dialog only showed the task field for a fixed list of four policy types (smolvla, pi0, pi0_fast, pi05), so other language-conditioned policies (groot, xvla, plugin policies, ...) had no way to pass a task.

Flip the check: requires_task is now true unless the policy type is known to ignore the task (act, diffusion, vqbet, tdmpc, sac). An unused task string is harmless, while a missing one makes a language-conditioned policy unusable. The field stays optional in the dialog.

Backend only, so no frontend/dist change. Added tests for the task-free, language-conditioned and unknown/missing type cases.

🤖 Generated with Claude Code

@nicolas-rabault nicolas-rabault left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Thanks for the follow-up!

@nicolas-rabault
nicolas-rabault merged commit 8ed1515 into huggingface:main Oct 29, 2026
2 checks 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.

2 participants