Merge pull request #57 from neptune-ai/ms/fix-lock-is-not-owned

Fix lock is not owned error in error handling block
neptune-ai · Oct 22, 2024 · f4fc266 · f4fc266
2 parents ed497f5 + 00e5c3b
commit f4fc266
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/neptune_scale/core/components/sync_process.py b/src/neptune_scale/core/components/sync_process.py
@@ -482,7 +482,8 @@ def work(self) -> None:
                     self._last_put_seq_wait.notify_all()
         except Exception as e:
             self._errors_queue.put(e)
-            self._last_put_seq_wait.notify_all()
+            with self._last_put_seq_wait:
+                self._last_put_seq_wait.notify_all()
             self.interrupt()
             raise NeptuneSynchronizationStopped() from e