sqlite: delete_node: don't assume the key is cached in self._ids

Related iterative/dvc#9826
iterative · Oct 3, 2023 · f98617a · f98617a
1 parent cd9e41c
commit f98617a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sqltrie/sqlite/sqlite.py b/src/sqltrie/sqlite/sqlite.py
@@ -293,7 +293,7 @@ def has_node(self, key: TrieKey) -> bool:
 
     def delete_node(self, key: TrieKey):
         node = self._get_node(key)
-        del self._ids[key]
+        self._ids.pop(key, None)
         self._conn.execute(
             """
             DELETE FROM nodes WHERE id = ?