Way to supress tqdm printout?

[eeilon79]

Hi,

I'm trying to use your package as a part of larger algorithm and I'd really like to suppress the printout as it is not relevant for my purposes. The solution here (diverting the standard output of the system) does not seem to work.

[wangguanan]

I have the same question. Is there any solution?

[ArcaneEmergence]

You need to use the GitHub version, not the pip version. There is a flag called tqdm_flag. Set it to False:
e.g.
cluster_ids_x, cluster_centers = kmeans(X=x, num_clusters=num_clusters, distance='euclidean', device=torch.device('cuda:0'), tqdm_flag=False)

It will still print the initialization though "running kMeans on device". For that you need to comment out the line in the source code.