Skip to content

Commit b51a372

Browse files
committed
AL-000: Fixed confidence for TextEntity
1 parent 982c410 commit b51a372

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

labelbox/data/serialization/ndjson/objects.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -355,8 +355,8 @@ def to_common(self) -> TextEntity:
355355
def from_common(cls, text_entity: TextEntity,
356356
classifications: List[ClassificationAnnotation], name: str,
357357
feature_schema_id: Cuid, extra: Dict[str, Any],
358-
data: Union[ImageData,
359-
TextData], confidence: float) -> "NDTextEntity":
358+
data: Union[ImageData,TextData],
359+
confidence: Optional[float] = None) -> "NDTextEntity":
360360
return cls(location=Location(
361361
start=text_entity.start,
362362
end=text_entity.end,

tests/data/serialization/ndjson/test_text_entity.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
"tests/data/assets/ndjson/text_entity_without_confidence_import.json"
1111
])
1212
def test_text_entity_import(filename: str):
13-
with open('tests/data/assets/ndjson/text_entity_import.json', 'r') as file:
13+
with open(filename, 'r') as file:
1414
data = json.load(file)
1515
res = NDJsonConverter.deserialize(data).as_list()
1616
res = list(NDJsonConverter.serialize(res))

0 commit comments

Comments
 (0)