From 9ff9ac8b3fac59bd014b7d33dffcb31b52c4f4a6 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 8 Oct 2025 16:33:27 +0200
Subject: [PATCH] updates kuzu lock reproduction

---
 working_dir_error_replication/reader.py | 1 -
 working_dir_error_replication/writer.py | 4 ++--
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/working_dir_error_replication/reader.py b/working_dir_error_replication/reader.py
index 5c54b771b..4ef78ea06 100644
--- a/working_dir_error_replication/reader.py
+++ b/working_dir_error_replication/reader.py
@@ -6,7 +6,6 @@ from cognee.infrastructure.databases.graph.kuzu.adapter import KuzuAdapter
 
 
 async def main():
-    time.sleep(20)
     adapter = KuzuAdapter("test.db")
     result = await adapter.query("MATCH (n:Node) RETURN COUNT(n)")
 
diff --git a/working_dir_error_replication/writer.py b/working_dir_error_replication/writer.py
index fb358b886..07f2cd276 100644
--- a/working_dir_error_replication/writer.py
+++ b/working_dir_error_replication/writer.py
@@ -20,14 +20,14 @@ def create_node(name):
 
 async def main():
     adapter = KuzuAdapter("test.db")
-    nodes = [create_node(f"Node{i}") for i in range(200000)]
+    nodes = [create_node(f"Node{i}") for i in range(2)]
 
     print("Writer: Starting...")
     await adapter.add_nodes(nodes)
 
     print("writer finished...")
 
-    time.sleep(10)
+    time.sleep(5)
 
 
 if __name__ == "__main__":