format

Author: farkob

labelbox/schema/project.py

@@ -851,13 +851,12 @@ def batches(self) -> PaginatedCollection:
             project(where: {id: $%s}) {id
             batches(after: $from, first: $first) { nodes { %s } pageInfo { endCursor }}}}
         """ % (id_param, id_param, query.results_query_part(Entity.Batch))
-        return PaginatedCollection(self.client,
-                                   query_str, {id_param: self.uid},
-                                   ['project', 'batches', 'nodes'],
-                                   Entity.Batch,
-                                   cursor_path=['project', 'batches',
-                                                'pageInfo', 'endCursor'],
-                                   experimental=True)
+        return PaginatedCollection(
+            self.client,
+            query_str, {id_param: self.uid}, ['project', 'batches', 'nodes'],
+            Entity.Batch,
+            cursor_path=['project', 'batches', 'pageInfo', 'endCursor'],
+            experimental=True)
 
     def upload_annotations(
             self,

tests/integration/test_batch.py

@@ -26,6 +26,7 @@ def test_create_batch(configured_project: Project, big_dataset: Dataset):
     assert batch.name == 'test-batch'
     assert batch.size == len(data_rows)
 
+
 def test_export_data_rows(configured_project: Project, dataset: Dataset):
     n_data_rows = 5
     task = dataset.create_data_rows([