Skip to content

Conversation

@Arun-Niranjan
Copy link
Collaborator

@Arun-Niranjan Arun-Niranjan commented Nov 30, 2025

Addresses #277 by allowing the Env class to output a categorical distribution for observations

This PR is stacked on top of #326 which should be reviewed and landed first

@conorheins
Copy link
Collaborator

Hi @Arun-Niranjan once your first PR lands, I'm going to ask that we delay dealing with this until the env refactor (see Issue #273) gets merged in -- that will have a lot of changes to the entire env API (the way we construct, reset, and step environments -- now using a separate env_params pytree and manual vmap calls on the reset and step methods), so the changes here will conflict with that. I think once that's merged in it (ASAP), we can re-introduce this PR but fresh on top of the new way the env is written

@Arun-Niranjan Arun-Niranjan force-pushed the fix/277-rollout-with-categorical-obs branch from 3875104 to 6a1e0d3 Compare December 7, 2025 20:24
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