From 2531e50a569c5db5e7422bdc5712e8dddcddbeff Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Thu, 11 Dec 2025 11:47:02 +0100
Subject: [PATCH] fix linting

---
 .../graph_completion_retriever_context_extension_test.py        | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cognee/tests/unit/modules/retrieval/graph_completion_retriever_context_extension_test.py b/cognee/tests/unit/modules/retrieval/graph_completion_retriever_context_extension_test.py
index d6696ff38..6a9b07d38 100644
--- a/cognee/tests/unit/modules/retrieval/graph_completion_retriever_context_extension_test.py
+++ b/cognee/tests/unit/modules/retrieval/graph_completion_retriever_context_extension_test.py
@@ -59,7 +59,7 @@ async def test_init_custom_params():
     assert retriever.user_prompt_path == "custom_user.txt"
     assert retriever.system_prompt_path == "custom_system.txt"
     assert retriever.system_prompt == "Custom prompt"
-    assert retriever.node_type == str
+    assert retriever.node_type is str
     assert retriever.node_name == ["node1"]
     assert retriever.save_interaction is True
     assert retriever.wide_search_top_k == 200