private static void addIDsExtension( PMML pmml, String key, RDD<Tuple2<Object, double[]>> features) { List<String> ids = fromRDD(features).keys().map(Functions.toStringValue()).collect(); PMMLUtils.addExtensionContent(pmml, key, ids); }