clean up and add conditions=True case #582

Job	Run time
test (ubuntu-latest, 3.10, jax)	1m 47s
test (ubuntu-latest, 3.10, numpy)	1m 49s
test (ubuntu-latest, 3.10, tensorflow)	1m 54s
test (ubuntu-latest, 3.10, torch)	1m 53s
test (ubuntu-latest, 3.11, jax)	1m 53s
test (ubuntu-latest, 3.11, numpy)	1m 37s
test (ubuntu-latest, 3.11, tensorflow)	1m 42s
test (ubuntu-latest, 3.11, torch)	1m 52s
test (windows-latest, 3.10, jax)	1m 40s
test (windows-latest, 3.10, numpy)	1m 33s
test (windows-latest, 3.10, tensorflow)	1m 40s
test (windows-latest, 3.10, torch)	1m 39s
test (windows-latest, 3.11, jax)	1m 38s
test (windows-latest, 3.11, numpy)	1m 40s
test (windows-latest, 3.11, tensorflow)	1m 38s
test (windows-latest, 3.11, torch)	1m 37s
	27m 32s

Provide feedback