def main(): # get dataset for artist sting db = DatabaseWrapper() cluster = ClusterModule() dataset = db.getStationSetForArtist("Coldplay") output = cluster.cluster(dataset)