Use `.iterator()` from peewee when querying

Closes #15 (closed)

worker_generic_training_dataset/db.py

@@ -17,13 +17,11 @@ def retrieve_element(element_id: UUID) -> Element:
 
 
 def list_classifications(element_id: UUID):
-    return (
-        Classification.select().where(Classification.element == element_id).iterator()
-    )
+    return Classification.select().where(Classification.element == element_id)
 
 
 def list_transcriptions(element_id: UUID):
-    return Transcription.select().where(Transcription.element == element_id).iterator()
+    return Transcription.select().where(Transcription.element == element_id)
 
 
 def list_transcription_entities(transcription_id: UUID):
@@ -32,5 +30,4 @@ def list_transcription_entities(transcription_id: UUID):
         .where(TranscriptionEntity.transcription == transcription_id)
         .join(Entity, on=TranscriptionEntity.entity)
         .join(EntityType, on=Entity.type)
-        .iterator()
     )