Tests

Add squeeze to adapter (#289) #419

Job	Run time
lint	1m 14s
test (ubuntu-latest, 3.10, jax)	8m 16s
test (ubuntu-latest, 3.10, tensorflow)	9m 25s
test (ubuntu-latest, 3.10, torch)	6m 1s
test (ubuntu-latest, 3.11, jax)	8m 54s
test (ubuntu-latest, 3.11, tensorflow)	9m 53s
test (ubuntu-latest, 3.11, torch)	6m 0s
test (windows-latest, 3.10, jax)	13m 29s
test (windows-latest, 3.10, tensorflow)	16m 16s
test (windows-latest, 3.10, torch)	13m 58s
test (windows-latest, 3.11, jax)	15m 59s
test (windows-latest, 3.11, tensorflow)	17m 38s
test (windows-latest, 3.11, torch)	14m 39s
	2h 21m 42s

Provide feedback