fix cv ranker

Author: ganik

src/python/nimbusml/internal/utils/data_roles.py

@@ -81,6 +81,20 @@ def to_attribute(role, suffix="_column_name"):
             return "row_id" + suffix
         return role.lower() + suffix
 
+    @staticmethod
+    def to_parameter(role, suffix="ColumnName"):
+        """
+        Converts a role into (as per manifesrt.json) parameter name.
+        ``GroupId --> RowGroupColumnName``.
+        """
+        if not isinstance(role, str):
+            raise TypeError("Unexpected role '{0}'".format(role))
+        if role == "Weight":
+            return "ExampleWeight" + suffix
+        if role == "GroupId":
+            return "RowGroup" + suffix
+        return role + suffix
+
     @staticmethod
     def to_role(column_name, suffix="_column_name"):
         """

src/python/nimbusml/pipeline.py

@@ -924,17 +924,17 @@ def process_input_output(classname, node, input_schema):
             else:
                 assigned = []
                 for role in sorted(DataRoles._allowed):
-                    attr = role + 'Column'
+                    attr = DataRoles.to_parameter(role)
                     if attr in inp:
                         assigned.append(inp[attr])
                 assigned = set(assigned)
                 not_assigned = [
                     col for col in input_schema if col not in assigned]
 
                 for role in sorted(DataRoles._allowed):
-                    attr = role + 'Column'
+                    attr = DataRoles.to_parameter(role)
                     if attr in inp:
-                        if attr == 'FeatureColumn' and inp[attr]\
+                        if attr == 'FeatureColumnName' and inp[attr]\
                                 not in input_schema:
                             val = not_assigned
                         else: