# Transition Program Proposal

Write a Python function `sample_next_state(state, action, rng)` for the task-specific API.
Do not rely on any benchmark or task name; use only the provided interface, schema, and examples.