chore: ruff format

cognee/tests/unit/modules/retrieval/graph_completion_retriever_context_extension_test.py

@@ -11,6 +11,7 @@ from cognee.modules.retrieval.graph_completion_context_extension_retriever impor
     GraphCompletionContextExtensionRetriever,
 )
 
+
 class TestGraphCompletionWithContextExtensionRetriever:
     @pytest.mark.asyncio
     async def test_graph_completion_extension_context_simple(self):

cognee/tests/unit/modules/retrieval/graph_completion_retriever_cot_test.py

@@ -9,6 +9,7 @@ from cognee.modules.graph.utils import resolve_edges_to_text
 from cognee.tasks.storage import add_data_points
 from cognee.modules.retrieval.graph_completion_cot_retriever import GraphCompletionCotRetriever
 
+
 class TestGraphCompletionCoTRetriever:
     @pytest.mark.asyncio
     async def test_graph_completion_cot_context_simple(self):

cognee/tests/unit/modules/retrieval/graph_completion_retriever_test.py

@@ -9,6 +9,7 @@ from cognee.modules.graph.utils import resolve_edges_to_text
 from cognee.tasks.storage import add_data_points
 from cognee.modules.retrieval.graph_completion_retriever import GraphCompletionRetriever
 
+
 class TestGraphCompletionRetriever:
     @pytest.mark.asyncio
     async def test_graph_completion_context_simple(self):

cognee/tests/unit/modules/retrieval/rag_completion_retriever_test.py

@@ -26,6 +26,7 @@ class DocumentChunkWithEntities(DataPoint):
 
     metadata: dict = {"index_fields": ["text"]}
 
+
 class TestRAGCompletionRetriever:
     @pytest.mark.asyncio
     async def test_rag_completion_context_simple(self):

cognee/tests/unit/modules/retrieval/temporal_retriever_test.py

@@ -139,6 +139,7 @@ async def test_filter_top_k_events_error_handling():
     with pytest.raises((KeyError, TypeError)):
         await tr.filter_top_k_events([{}], [])
 
+
 class _FakeRetriever(TemporalRetriever):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)