forest_util.py norm: Remove usage of list
Always use JAX arrays when calling JAX's norm implementation. This resolves an issue with the most recent version of JAX.
Always use JAX arrays when calling JAX's norm implementation. This resolves an issue with the most recent version of JAX.
Merge details
requested review from @pfrank
assigned to @pfrank
This should be merged soon to fix the failing pipeline.
merged
mentioned in commit dd6a52f6