Address pr comments

Author: Gareth

labelbox/schema/project.py

@@ -35,7 +35,7 @@
 
 logger = logging.getLogger(__name__)
 
-MAX_BATCH_SIZE = 1000
+MAX_QUEUE_BATCH_SIZE = 1000
 
 
 class QueueMode(enum.Enum):
@@ -467,8 +467,8 @@ def _post_batch(self, method, data_row_ids: List[str]):
         if self.queue_mode() != QueueMode.Batch:
             raise ValueError("Project must be in batch mode")
 
-        if len(data_row_ids) > MAX_BATCH_SIZE:
-            raise ValueError(f"Batch exceeds max size of {MAX_BATCH_SIZE}, consider breaking it into parts")
+        if len(data_row_ids) > MAX_QUEUE_BATCH_SIZE:
+            raise ValueError(f"Batch exceeds max size of {MAX_QUEUE_BATCH_SIZE}, consider breaking it into parts")
 
         query = """mutation %sPyApi($projectId: ID!, $dataRowIds: [ID!]!) {
               project(where: {id: $projectId}) {
@@ -512,11 +512,8 @@ def _update_queue_mode(self, mode: QueueMode) -> QueueMode:
         query_str = """mutation %s($projectId: ID!, $status: TagSetStatusInput!) {
               project(where: {id: $projectId}) {
                  setTagSetStatus(input: {tagSetStatus: $status}) {
-                    id
                     tagSetStatus
-                    __typename
                 }
-                __typename
             }
         }       
         """ % "setTagSetStatusPyApi"
@@ -532,9 +529,7 @@ def queue_mode(self):
 
         query_str = """query %s($projectId: ID!) {
               project(where: {id: $projectId}) {
-                 id
                  tagSetStatus
-                __typename
             }
         }       
         """ % "GetTagSetStatusPyApi"

tests/data/annotation_types/geometry/test_rectangle.py

@@ -15,8 +15,8 @@ def test_rectangle():
     expected['coordinates'] = tuple([tuple([tuple(x) for x in points[0]])])
     assert rectangle.shapely.__geo_interface__ == expected
 
-    # raster = rectangle.draw(height=32, width=32)
-    # assert (cv2.imread("tests/data/assets/rectangle.png") == raster).all()
+    raster = rectangle.draw(height=32, width=32)
+    assert (cv2.imread("tests/data/assets/rectangle.png") == raster).all()
 
     xyhw = Rectangle.from_xyhw(0., 0, 10, 10)
     assert xyhw.start == Point(x=0, y=0.)

tests/integration/test_project.py

@@ -111,5 +111,4 @@ def test_queued_data_row_export(configured_project):
 def test_queue_mode(configured_project: Project):
     assert configured_project.queue_mode() == QueueMode.Dataset
     configured_project.update(queue_mode=QueueMode.Batch)
-    # TODO: understand why this fails
     assert configured_project.queue_mode() == QueueMode.Batch