format

Author: gdj0nes

labelbox/schema/data_row_metadata.py

@@ -53,6 +53,7 @@ def id(self):
 
 
 class _CamelCaseMixin(BaseModel):
+
     class Config:
         allow_population_by_field_name = True
         alias_generator = camel_case
@@ -135,14 +136,18 @@ def _build_ontology(self):
             f for f in self.fields if f.reserved
         ]
         self.reserved_by_id = self._make_id_index(self.reserved_fields)
-        self.reserved_by_name: Dict[str, DataRowMetadataSchema] = self._make_name_index(self.reserved_fields)
+        self.reserved_by_name: Dict[
+            str,
+            DataRowMetadataSchema] = self._make_name_index(self.reserved_fields)
 
         # custom fields
         self.custom_fields: List[DataRowMetadataSchema] = [
             f for f in self.fields if not f.reserved
         ]
         self.custom_by_id = self._make_id_index(self.custom_fields)
-        self.custom_by_name: Dict[str, DataRowMetadataSchema] = self._make_name_index(self.custom_fields)
+        self.custom_by_name: Dict[
+            str,
+            DataRowMetadataSchema] = self._make_name_index(self.custom_fields)
 
     @staticmethod
     def _make_name_index(fields: List[DataRowMetadataSchema]):
@@ -158,7 +163,7 @@ def _make_name_index(fields: List[DataRowMetadataSchema]):
 
     @staticmethod
     def _make_id_index(
-            fields: List[DataRowMetadataSchema]
+        fields: List[DataRowMetadataSchema]
     ) -> Dict[SchemaId, DataRowMetadataSchema]:
         index = {}
         for f in fields:
@@ -209,9 +214,9 @@ def _parse_ontology(raw_ontology) -> List[DataRowMetadataSchema]:
         return fields
 
     def parse_metadata(
-            self, unparsed: List[Dict[str,
-                                      List[Union[str,
-                                                 Dict]]]]) -> List[DataRowMetadata]:
+        self, unparsed: List[Dict[str,
+                                  List[Union[str,
+                                             Dict]]]]) -> List[DataRowMetadata]:
         """ Parse metadata responses
 
         >>> mdo.parse_metadata([metdata])
@@ -272,7 +277,7 @@ def bulk_upsert(
             raise ValueError("Empty list passed")
 
         def _batch_upsert(
-                upserts: List[_UpsertBatchDataRowMetadata]
+            upserts: List[_UpsertBatchDataRowMetadata]
         ) -> List[DataRowMetadataBatchResponse]:
             query = """mutation UpsertDataRowMetadataBetaPyApi($metadata: [DataRowCustomMetadataBatchUpsertInput!]!) {
                 upsertDataRowCustomMetadata(data: $metadata){
@@ -305,13 +310,13 @@ def _batch_upsert(
                     fields=list(
                         chain.from_iterable(
                             self._parse_upsert(m) for m in m.fields))).dict(
-                    by_alias=True))
+                                by_alias=True))
 
         res = _batch_operations(_batch_upsert, items, self._batch_size)
         return res
 
     def bulk_delete(
-            self, deletes: List[DeleteDataRowMetadata]
+        self, deletes: List[DeleteDataRowMetadata]
     ) -> List[DataRowMetadataBatchResponse]:
         """ Delete metadata from a datarow by specifiying the fields you want to remove
 
@@ -338,7 +343,7 @@ def bulk_delete(
             raise ValueError("Empty list passed")
 
         def _batch_delete(
-                deletes: List[_DeleteBatchDataRowMetadata]
+            deletes: List[_DeleteBatchDataRowMetadata]
         ) -> List[DataRowMetadataBatchResponse]:
             query = """mutation DeleteDataRowMetadataBetaPyApi($deletes: [DataRowCustomMetadataBatchDeleteInput!]!) {
                 deleteDataRowCustomMetadata(data: $deletes) {
@@ -459,9 +464,9 @@ def _batch_items(iterable: List[Any], size: int) -> Generator[Any, None, None]:
 
 
 def _batch_operations(
-        batch_function: _BatchFunction,
-        items: List,
-        batch_size: int = 100,
+    batch_function: _BatchFunction,
+    items: List,
+    batch_size: int = 100,
 ):
     response = []
 

tests/integration/test_data_row_metadata.py

@@ -36,11 +36,11 @@ def mdo(client):
 @pytest.fixture
 def big_dataset(dataset: Dataset, image_url):
     task = dataset.create_data_rows([
-                                        {
-                                            "row_data": image_url,
-                                            "external_id": "my-image"
-                                        },
-                                    ] * 250)
+        {
+            "row_data": image_url,
+            "external_id": "my-image"
+        },
+    ] * 250)
     task.wait_till_done()
 
     yield dataset
@@ -50,11 +50,11 @@ def big_dataset(dataset: Dataset, image_url):
 def wait_for_embeddings_svc(data_row_ids, mdo):
     for idx in range(5):
         if all([
-            len(metadata.fields)
-            for metadata in mdo.bulk_export(data_row_ids)
+                len(metadata.fields)
+                for metadata in mdo.bulk_export(data_row_ids)
         ]):
             return
-        time.sleep((idx + 1) ** 2)
+        time.sleep((idx + 1)**2)
     raise Exception("Embedding svc failed to update metadata.")
 
 
@@ -90,16 +90,10 @@ def test_get_datarow_metadata_ontology(mdo):
                 schema_id=mdo.reserved_by_name["captureDateTime"].uid,
                 value=datetime.utcnow(),
             ),
-            DataRowMetadataField(
-                schema_id=split.parent,
-                value=split.uid
-            ),
-            DataRowMetadataField(
-                schema_id=mdo.reserved_by_name["tag"].uid,
-                value="hello-world"
-            ),
-        ]
-    )
+            DataRowMetadataField(schema_id=split.parent, value=split.uid),
+            DataRowMetadataField(schema_id=mdo.reserved_by_name["tag"].uid,
+                                 value="hello-world"),
+        ])
 
 
 def test_bulk_upsert_datarow_metadata(datarow, mdo: DataRowMetadataOntology):
@@ -248,22 +242,27 @@ def test_parse_raw_metadata(mdo):
     example = {
         'dataRowId':
             'ckr6kkfx801ui0yrtg9fje8xh',
-        'fields': [{
-            'schemaId': 'cko8s9r5v0001h2dk9elqdidh',
-            'value': 'my-new-message'
-        }, {
-            'schemaId': 'cko8sbczn0002h2dkdaxb5kal',
-            'value': {}
-        }, {
-            'schemaId': 'cko8sbscr0003h2dk04w86hof',
-            'value': {}
-        }, {
-            'schemaId': 'cko8sdzv70006h2dk8jg64zvb',
-            'value': '2021-07-20T21:41:14.606710Z'
-        }, {
-            'schemaId': FAKE_SCHEMA_ID,
-            'value': 0.5
-        },
+        'fields': [
+            {
+                'schemaId': 'cko8s9r5v0001h2dk9elqdidh',
+                'value': 'my-new-message'
+            },
+            {
+                'schemaId': 'cko8sbczn0002h2dkdaxb5kal',
+                'value': {}
+            },
+            {
+                'schemaId': 'cko8sbscr0003h2dk04w86hof',
+                'value': {}
+            },
+            {
+                'schemaId': 'cko8sdzv70006h2dk8jg64zvb',
+                'value': '2021-07-20T21:41:14.606710Z'
+            },
+            {
+                'schemaId': FAKE_SCHEMA_ID,
+                'value': 0.5
+            },
         ]
     }