From 9ee93a4260698a83ebf1fce992894b99fada8351 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Thu, 18 Sep 2025 18:25:00 +0200
Subject: [PATCH] ruff fix

---
 cognee/tasks/graph/extract_graph_from_data.py    | 8 ++++----
 cognee/tasks/graph/extract_graph_from_data_v2.py | 6 +++---
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/cognee/tasks/graph/extract_graph_from_data.py b/cognee/tasks/graph/extract_graph_from_data.py
index 5c3b11821..391c6fabe 100644
--- a/cognee/tasks/graph/extract_graph_from_data.py
+++ b/cognee/tasks/graph/extract_graph_from_data.py
@@ -29,20 +29,20 @@ async def integrate_chunk_graphs(
     ontology_resolver: BaseOntologyResolver,
 ) -> List[DocumentChunk]:
     """Integrate chunk graphs with ontology validation and store in databases.
-    
+
     This function processes document chunks and their associated knowledge graphs,
     validates entities against an ontology resolver, and stores the integrated
     data points and edges in the configured databases.
-    
+
     Args:
         data_chunks: List of document chunks containing source data
         chunk_graphs: List of knowledge graphs corresponding to each chunk
         graph_model: Pydantic model class for graph data validation
         ontology_resolver: Resolver for validating entities against ontology
-        
+
     Returns:
         List of updated DocumentChunk objects with integrated data
-        
+
     Raises:
         InvalidChunkGraphInputError: If input validation fails
         InvalidGraphModelError: If graph model validation fails
diff --git a/cognee/tasks/graph/extract_graph_from_data_v2.py b/cognee/tasks/graph/extract_graph_from_data_v2.py
index 0a8869784..8cea6602e 100644
--- a/cognee/tasks/graph/extract_graph_from_data_v2.py
+++ b/cognee/tasks/graph/extract_graph_from_data_v2.py
@@ -20,15 +20,15 @@ async def extract_graph_from_data(
     ontology_adapter: BaseOntologyResolver = None,
 ) -> List[DocumentChunk]:
     """Extract and update graph data from document chunks using cascade extraction.
-    
+
     This function performs multi-step graph extraction from document chunks,
     using cascade extraction techniques to build comprehensive knowledge graphs.
-    
+
     Args:
         data_chunks: List of document chunks to process
         n_rounds: Number of extraction rounds to perform (default: 2)
         ontology_adapter: Resolver for validating entities against ontology
-        
+
     Returns:
         List of updated DocumentChunk objects with extracted graph data
     """