KMeansTrainer: fix bug in py binding