Type Conflict Between JAX NumPy and Python's float() in EvoJAX SlimeVolley

[Yuta-Kondo]

I think there exists the type conflict of jax numpy and python's float() function.
In evojax/task/slimevolley.py, jax.numpy.float32 is used, and this is the tax scaler constructor of type float 32. But, Jax represents scalars as zero dimensional arrays, which causes the conflict with python's float() function.

When I ran examples/train_slimevolley.py file, I got this error after training the model.

TypeError: Only scalar arrays can be converted to Python scalars; got arr.ndim=1