Fix mongo batch record.end update

pennsignals · May 4, 2020 · aa0bc1c · aa0bc1c
1 parent 2cd64e3
commit aa0bc1c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/dsdk/mongo.py b/src/dsdk/mongo.py
@@ -106,9 +106,9 @@ def open_batch(
 
             yield batch
 
-            key, doc = batch.as_update_doc()
-            with self.open_mongo() as database:
-                update_one(database.batches, key, doc)
+        key, doc = batch.as_update_doc()
+        with self.open_mongo() as database:
+            update_one(database.batches, key, doc)
 
     def store_evidence(self, batch: Batch, *args, **kwargs) -> None:
         """Store Evidence."""