Support global keys for Project update_data_row_labeling_priority

Author: Val Brodsky

labelbox/schema/project.py

@@ -1185,10 +1185,26 @@ def set_labeling_parameter_overrides(self, data) -> bool:
         res = self.client.execute(query_str, {id_param: self.uid})
         return res["project"]["setLabelingParameterOverrides"]["success"]
 
+    @overload
+    def update_data_row_labeling_priority(
+        self,
+        data_rows: DataRowIdentifiers,
+        priority: int,
+    ) -> bool:
+        pass
+
+    @overload
     def update_data_row_labeling_priority(
         self,
         data_rows: List[str],
         priority: int,
+    ) -> bool:
+        pass
+
+    def update_data_row_labeling_priority(
+        self,
+        data_rows,
+        priority: int,
     ) -> bool:
         """
         Updates labeling parameter overrides to this project in bulk. This method allows up to 1 million data rows to be
@@ -1198,25 +1214,31 @@ def update_data_row_labeling_priority(
             https://docs.labelbox.com/en/configure-editor/queue-system#reservation-system
 
         Args:
-            data_rows (iterable): An iterable of data row ids.
+            data_rows: a list of data row ids to update priorities for. This can be a list of strings or a DataRowIdentifiers object 
+                DataRowIdentifier objects are lists of ids or global keys. A DataIdentifier object can be a UniqueIds or GlobalKeys class.
             priority (int): Priority for the new override. See above for more information.
 
         Returns:
             bool, indicates if the operation was a success.
         """
 
+        if isinstance(data_rows, list):
+            data_rows = UniqueIds(data_rows)
+            warnings.warn("Using data row ids will be deprecated. Please use "
+                          "UniqueIds or GlobalKeys instead.")
+
         method = "createQueuePriorityUpdateTask"
         priority_param = "priority"
         project_param = "projectId"
-        data_rows_param = "dataRowIds"
+        data_rows_param = "dataRowIdentifiers"
         query_str = """mutation %sPyApi(
               $%s: Int!
               $%s: ID!
-              $%s: [ID!]
+              $%s: QueuePriorityUpdateDataRowIdentifiersInput
             ) {
               project(where: { id: $%s }) {
                 %s(
-                  data: { priority: $%s, dataRowIds: $%s }
+                  data: { priority: $%s, dataRowIdentifiers: $%s }
                 ) {
                   taskId
                 }
@@ -1228,7 +1250,10 @@ def update_data_row_labeling_priority(
             query_str, {
                 priority_param: priority,
                 project_param: self.uid,
-                data_rows_param: data_rows
+                data_rows_param: {
+                    "ids": [id for id in data_rows],
+                    "idType": data_rows._id_type,
+                },
             })["project"][method]
 
         task_id = res['taskId']