simple kmeans test with n_jobs=2

km_test.py

from sklearn.cluster import KMeans
from sklearn.datasets import make_blobs
X, y = make_blobs()

KMeans(n_jobs=2).fit(X)
print("done")

km_test_main.py

from sklearn.cluster import KMeans
from sklearn.datasets import make_blobs


def main():
    X, y = make_blobs()

    KMeans(n_jobs=2).fit(X)
    print("done")


if __name__ == "__main__":
    main()