Error: Failed to serialize to YAML format: cannot represent an object: np.float64(0.0019339837881604581)

This object may contain types that are not YAML-serializable.