def save_batches(col: collection.Collection, batches):
"""Saves the batches to the specified collection
in the database"""
logging.debug(f'Saving documents to the {col.full_name} collection')
ids = col.insert_many([{
'_id':
i,
'data':
pickle.dumps(data, pickle.HIGHEST_PROTOCOL),
'labels':
pickle.dumps(labels, pickle.HIGHEST_PROTOCOL)
} for i, (data, labels) in enumerate(batches)]).inserted_ids
logging.debug(f'Inserted {len(ids)} documents')
