Skip to content

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

Open
ravediamond wants to merge 1 commit into
huggingface:mainfrom
ravediamond:inference-task-field
Open

ravediamond wants to merge 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

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