Revert "try to fix #5660"

This reverts commit 4445e54. Signed-off-by: Wenqi Li <wenqil@nvidia.com>
Project-MONAI · Dec 12, 2022 · d9709eb · d9709eb
1 parent 4445e54
commit d9709eb
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/monai/data/dataset.py b/monai/data/dataset.py
@@ -1068,8 +1068,7 @@ def set_data(self, data: Sequence):
         if self.shuffle:
             data = copy(data)
             self.randomize(data)
-        with self._update_lock:
-            super().set_data(data)
+        super().set_data(data)
 
     def randomize(self, data: Sequence) -> None:
         try:
@@ -1108,8 +1107,7 @@ def _restart(self):
         Restart background thread if killed for some reason.
 
         """
-        with self._update_lock:
-            self._round = 1
+        self._round = 1
         self._replace_mgr = threading.Thread(target=self.manage_replacement, daemon=True)
         self._replace_mgr.start()