Tests

Replace `dm-tree` with `optree` #1819

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Check the code format	1m 54s
Run tests (3.9, tensorflow)	10m 19s
Run tests (3.9, jax)	13m 37s
Run tests (3.9, torch)	9m 26s
Run tests (3.9, numpy)	4m 32s
	39m 48s