Skip to content

Conversation

felixzinn
Copy link
Contributor

visualization of pytrees is now possible with the standalone treescope package. This PR adapts to this change.
Furthermore penzai moved to a new V2 API, so a minor change to the visualization had to be made to accomodate to this.

Copy link
Contributor Author

@felixzinn felixzinn left a comment

Choose a reason for hiding this comment

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

a __treescope_repr__ for evermore components has been implemented which removes penzai as a dependency. Visualization of evermore components is now done with the standalone package treescope.

@felixzinn felixzinn marked this pull request as ready for review August 14, 2024 06:44
@pfackeldey pfackeldey self-assigned this Aug 14, 2024
@pfackeldey
Copy link
Owner

awesome work!! 🎉

@pfackeldey pfackeldey merged commit 732354e into pfackeldey:main Aug 14, 2024
3 checks passed
@felixzinn felixzinn deleted the visualization-treescope branch August 14, 2024 11:27
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