Tests

Adapter: Minor fixes and more thorough testing #481

Job	Run time
lint	1m 24s
test (ubuntu-latest, 3.10, jax)	8m 7s
test (ubuntu-latest, 3.10, tensorflow)	20m 57s
test (ubuntu-latest, 3.10, torch)	10m 38s
test (ubuntu-latest, 3.11, jax)	9m 9s
test (ubuntu-latest, 3.11, tensorflow)	23m 19s
test (ubuntu-latest, 3.11, torch)	11m 9s
test (windows-latest, 3.10, jax)	15m 37s
test (windows-latest, 3.10, tensorflow)	31m 43s
test (windows-latest, 3.10, torch)	23m 42s
test (windows-latest, 3.11, jax)	19m 48s
test (windows-latest, 3.11, tensorflow)	33m 34s
test (windows-latest, 3.11, torch)	26m 11s
	3h 55m 18s

Provide feedback