import numpy as np


def kmeans(
    X_train: np.array,
    y_train: np.array,
    args: dict,
):
    # for this alg, the only argument
    # is the number of clusters, k
    # and max iterations

    return