Fix floating point subdtype check in `kl_energies`

np.float was deprecated in Numpy 1.20.

The general class of floating point dtypes is np.floating, so check for that.