Fix BaseReview.review bug for ReviewSimulate (#1529)

asreview · Sep 21, 2023 · b22ec6d · b22ec6d
1 parent 0bbf7a2
commit b22ec6d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/asreview/review/base.py b/asreview/review/base.py
@@ -194,7 +194,6 @@ def review(self):
             # write to state when stopped
             pbar_rel.close()
             pbar_total.close()
-            self._write_to_state()
 
     def _label_priors(self):
         """Make sure the prior records are labeled."""

diff --git a/asreview/review/simulate.py b/asreview/review/simulate.py
@@ -226,6 +226,10 @@ def _label_priors(self):
             pending_labels = self.data_labels[pending]
             state.add_labeling_data(pending, pending_labels)
 
+    def review(self):
+        super().review()
+        self._write_to_state()
+
     def _stop_review(self):
         """In simulation mode, the stop review function should get the labeled
         records list from the reviewer attribute."""