Add accessors for id_type in Identiables

Author: Val Brodsky

labelbox/schema/data_row_metadata.py

@@ -656,7 +656,7 @@ def _bulk_export(
                     query, {
                         "dataRowIdentifiers": {
                             "ids": [id for id in _data_row_ids],
-                            "idType": _data_row_ids._id_type
+                            "idType": _data_row_ids.id_type
                         }
                     })['dataRowCustomMetadata'])
 

labelbox/schema/identifiables.py

@@ -14,6 +14,10 @@ def __init__(self, iterable, id_type: IdType):
         self._iterable = iterable
         self._id_type = id_type
 
+    @property
+    def id_type(self):
+        return self._id_type
+
     def __iter__(self):
         return iter(self._iterable)
 

labelbox/schema/project.py

@@ -1289,7 +1289,7 @@ def update_data_row_labeling_priority(
                 project_param: self.uid,
                 data_rows_param: {
                     "ids": [id for id in data_rows],
-                    "idType": data_rows._id_type,
+                    "idType": data_rows.id_type,
                 },
             })["project"][method]
 
@@ -1484,7 +1484,7 @@ def move_data_rows_to_task_queue(self, data_row_ids, task_queue_id: str):
                 "queueId": task_queue_id,
                 "dataRowIdentifiers": {
                     "ids": [id for id in data_row_ids],
-                    "idType": data_row_ids._id_type,
+                    "idType": data_row_ids.id_type,
                 },
             },
             timeout=180.0,

tests/integration/test_data_row_metadata.py

@@ -103,12 +103,12 @@ def test_bulk_export_datarow_metadata(data_row, mdo: DataRowMetadataOntology):
     assert exported[0].data_row_id == data_row.uid
     assert len([field for field in exported[0].fields]) == 3
 
-    exported = mdo.bulk_export(UniqueIds(data_row.uid))
+    exported = mdo.bulk_export(UniqueIds([data_row.uid]))
     assert exported[0].global_key == data_row.global_key
     assert exported[0].data_row_id == data_row.uid
     assert len([field for field in exported[0].fields]) == 3
 
-    exported = mdo.bulk_export(GlobalKeys(data_row.uid))
+    exported = mdo.bulk_export(GlobalKeys([data_row.global_key]))
     assert exported[0].global_key == data_row.global_key
     assert exported[0].data_row_id == data_row.uid
     assert len([field for field in exported[0].fields]) == 3

tests/unit/test_unit_identifiables.py

@@ -5,15 +5,15 @@ def test_unique_ids():
     ids = ["a", "b", "c"]
     identifiables = UniqueIds(ids)
     assert [i for i in identifiables] == ids
-    assert identifiables._id_type == "ID"
+    assert identifiables.id_type == "ID"
     assert len(identifiables) == 3
 
 
 def test_global_keys():
     ids = ["a", "b", "c"]
     identifiables = GlobalKeys(ids)
     assert [i for i in identifiables] == ids
-    assert identifiables._id_type == "GKEY"
+    assert identifiables.id_type == "GKEY"
     assert len(identifiables) == 3