fix kmeans broadcast error caused by parakeet