Tests

Add squeeze to adapter #417

Job	Run time
lint	1m 39s
test (ubuntu-latest, 3.10, jax)	8m 22s
test (ubuntu-latest, 3.10, tensorflow)	9m 9s
test (ubuntu-latest, 3.10, torch)	6m 3s
test (ubuntu-latest, 3.11, jax)	9m 25s
test (ubuntu-latest, 3.11, tensorflow)	9m 49s
test (ubuntu-latest, 3.11, torch)	6m 4s
test (windows-latest, 3.10, jax)	14m 13s
test (windows-latest, 3.10, tensorflow)	15m 45s
test (windows-latest, 3.10, torch)	12m 45s
test (windows-latest, 3.11, jax)	15m 19s
test (windows-latest, 3.11, tensorflow)	17m 29s
test (windows-latest, 3.11, torch)	14m 5s
	2h 20m 7s

Provide feedback