def python_update_assignments(X, centroids):
  dists = [[np.sum( (x-y) ** 2) for y in centroids] for x in X]
  return np.argmin(dists, 1)
dataset/ETHPy150Open iskandr/parakeet/test/algorithms/test_kmeans.py/python_update_assignments/original