diff --git a/unstructured/staging/datasaur.py b/unstructured/staging/datasaur.py index a7f96ae1cc..c73401d7b0 100644 --- a/unstructured/staging/datasaur.py +++ b/unstructured/staging/datasaur.py @@ -8,8 +8,6 @@ def stage_for_datasaur( entities: Optional[List[List[Dict[str, Any]]]] = None, ) -> List[Dict[str, Any]]: """Convert a list of elements into a list of dictionaries for use in Datasaur""" - result: List[Dict[str, Any]] = [] - _entities: List[List[Dict[str, Any]]] = [[] for _ in range(len(elements))] if entities is not None: if len(entities) != len(elements): @@ -21,11 +19,7 @@ def stage_for_datasaur( _entities = entities - for i, item in enumerate(elements): - data = {"text": item.text, "entities": _entities[i]} - result.append(data) - - return result + return [{"text": item.text, "entities": _entities[i]} for i, item in enumerate(elements)] def _validate_datasaur_entity(entity: Dict[str, Any]):