Skip to content

Cache endpoints in InterpolatedUnivariateSpline#116

Open
allen-adastra wants to merge 2 commits intoDifferentiableUniverseInitiative:masterfrom
allen-adastra:fix_interpolate_tree_flatten
Open

Cache endpoints in InterpolatedUnivariateSpline#116
allen-adastra wants to merge 2 commits intoDifferentiableUniverseInitiative:masterfrom
allen-adastra:fix_interpolate_tree_flatten

Conversation

@allen-adastra
Copy link

While trying to use tree_map on InterpolatedUnivariateSpline, I noticed endpoints weren't properly cached.

One line fix + add a unit test to make sure we can operate InterpolatedUnivariateSpline on pytrees.

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