Tests

Remove `bool` option for `scale_cost` #1804

Job	Run time
Fast GPU tests Python 3.10 on ubuntu-22.04	18m 21s
Python 3.8 on ubuntu-latest	45m 32s
Fast tests Python 3.9 jax-default	10m 22s
Python 3.10 on ubuntu-latest	1h 1m 14s
Fast tests Python 3.9 jax-latest	10m 17s
Python 3.11 on ubuntu-latest	1h 8m 6s
Python 3.12 on ubuntu-latest	1h 14m 1s
Python 3.9 on macos-latest	1h 41m 58s
Python 3.10 on macos-14	44m 13s
	7h 14m 4s

Provide feedback