Skip to content

Commit 7b9f8b4

Browse files
author
gdj0nes
committed
FIX: type conversion
1 parent 11f26a4 commit 7b9f8b4

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

labelbox/schema/data_row_metadata.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -84,16 +84,10 @@ class DataRowMetadataBatchResponse(_CamelCaseMixin):
8484
# Don't want to crowd the name space with internals
8585

8686

87-
# Bulk upsert values
88-
class _UpsertDataRowMetadataInput(_CamelCaseMixin):
89-
schema_id: str
90-
value: Any
91-
92-
9387
# Batch of upsert values for a datarow
9488
class _UpsertBatchDataRowMetadata(_CamelCaseMixin):
9589
data_row_id: str
96-
fields: List[_UpsertDataRowMetadataInput]
90+
fields: List[Dict]
9791

9892

9993
class _DeleteBatchDataRowMetadata(_CamelCaseMixin):
@@ -404,7 +398,7 @@ def _bulk_export(_data_row_ids: List[str]) -> List[DataRowMetadata]:
404398

405399
def _parse_upsert(
406400
self, metadatum: DataRowMetadataField
407-
) -> List[_UpsertDataRowMetadataInput]:
401+
) -> List[Dict]:
408402
"""Format for metadata upserts to GQL"""
409403

410404
if metadatum.schema_id not in self.fields_by_id:
@@ -428,7 +422,7 @@ def _parse_upsert(
428422
else:
429423
raise ValueError(f"Unknown type: {schema}")
430424

431-
return [_UpsertDataRowMetadataInput(**p) for p in parsed]
425+
return parsed
432426

433427
def _validate_delete(self, delete: DeleteDataRowMetadata):
434428
if not len(delete.fields):

0 commit comments

Comments
 (0)