fix rm_record bug using new write_lock

pepkit · Mar 14, 2024 · fac8a6b · fac8a6b
1 parent a46ed1e
commit fac8a6b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pipestat/backends/file_backend/filebackend.py b/pipestat/backends/file_backend/filebackend.py
@@ -311,8 +311,8 @@ def remove_record(
             try:
                 _LOGGER.info(f"Removing '{record_identifier}' record")
                 del self._data[self.pipeline_name][self.pipeline_type][record_identifier]
-                with self._data as locked_data:
-                    locked_data.write()
+                with write_lock(self._data) as data_locked:
+                    data_locked.write()
                 return True
             except:
                 _LOGGER.warning(