From bacfd1de06566bd029530e24aaec6680518db764 Mon Sep 17 00:00:00 2001 From: Daniel O'Connell Date: Sun, 17 Dec 2023 23:14:43 +0100 Subject: [PATCH] refresh session --- align_data/common/alignment_dataset.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/align_data/common/alignment_dataset.py b/align_data/common/alignment_dataset.py index 06da73d..91db5ac 100644 --- a/align_data/common/alignment_dataset.py +++ b/align_data/common/alignment_dataset.py @@ -109,9 +109,9 @@ def commit() -> bool: session.rollback() return False - with make_session() as session: - items = iter(entries) - while batch := tuple(islice(items, self.batch_size)): + items = iter(entries) + while batch := tuple(islice(items, self.batch_size)): + with make_session() as session: self._add_batch(session, batch) # there might be duplicates in the batch, so if they cause # an exception, try to commit them one by one