Add Project.upsert_review_queue

Author: florijanstamenkovic

labelbox/schema/project.py

@@ -225,6 +225,21 @@ def unset_labeling_parameter_overrides(self, data_rows):
         res = self.client.execute(query_str, {project_param: self.uid})
         return res["data"]["project"]["unsetLabelingParameterOverrides"]["success"]
 
+    def upsert_review_queue(self, quota_factor):
+        """ Reinitiate the review queue for this project.
+        Args:
+            quota_factor (float): Which part (percentage) of the queue
+                to reinitiate. Between 0 and 1.
+        """
+        project_param = "projectId"
+        quota_param = "quotaFactor"
+        query_str = """mutation UpsertReviewQueuePyApi($%s: ID!, $%s: Float!){
+            upsertReviewQueue(where:{project: {id: $%s}}
+                            data:{quotaFactor: $%s}) {id}}""" % (
+            project_param, quota_param, project_param, quota_param)
+        res = self.client.execute(
+            query_str, {project_param: self.uid, quota_param: quota_factor})
+
 
 class LabelingParameterOverride(DbObject):
     priority = Field.Int("priority")

tests/integration/test_project.py

@@ -50,3 +50,9 @@ def test_project_filtering(client, rand_gen):
 
     p1.delete()
     p2.delete()
+
+
+def test_upsert_review_queue(client, rand_gen):
+    project = client.create_project(name=rand_gen(str))
+    project.upsert_review_queue(0.6)
+    project.delete()