env.render support
- Introduced a unified rendering API covering
rgb_array,ansi,svgand adetailedmode for step visualizations. - Added
StepVisualizationWrapperwhich renders before/after transitions, action metadata and highlights changed cells. - Overhauled wrapper organization: action and observation wrappers refactored and consolidated under
jaxarc.wrappers. - Documentation updated to showcase the new rendering APIs and wrapper import points.
What’s new
-
Unified Render API
env.render(state, mode="rgb_array")-> returns numpy array suitable for saving as an image.env.render(state, mode="ansi")-> returns a string to print to terminal (uses rich for color fidelity).env.render(state, mode="svg")-> returns SVG string for embedding in notebooks and docs.StepVisualizationWrappersupportsmode="detailed"to produce annotated SVG with changes and action metadata.
-
Wrapper Consolidation
- Import wrappers from
jaxarc.wrappers(e.g.,from jaxarc.wrappers import PointActionWrapper). jaxarc.envsnow re-exports wrapper classes for backward compatibility, butjaxarc.wrappersis the recommended import.
- Import wrappers from
-
Docs
- Notebook examples updated under
docs/tutorialsto illustrate the env render modes andStepVisualizationWrapper. docs/api/wrappers.mdanddocs/tutorials/visualizing-tasks.ipynbinclude new sections for environment rendering.
- Notebook examples updated under
Full Changelog: v1.0.1...v1.0.2