Ruff formatting

Author: NathalieCharbel

src/neo4j_graphrag/experimental/components/entity_relation_extractor.py

@@ -19,7 +19,7 @@
 import enum
 import json
 import logging
-from typing import Any, List, Optional, Union, cast, Dict
+from typing import Any, Dict, List, Optional, Union, cast
 
 import json_repair
 from pydantic import ValidationError, validate_call
@@ -33,9 +33,9 @@
     Neo4jGraph,
     Neo4jNode,
     Neo4jRelationship,
+    SchemaEnforcementMode,
     TextChunk,
     TextChunks,
-    SchemaEnforcementMode,
 )
 from neo4j_graphrag.experimental.pipeline.component import Component
 from neo4j_graphrag.experimental.pipeline.exceptions import InvalidJSONError

src/neo4j_graphrag/experimental/pipeline/config/template_pipeline/simple_kg_builder.py

@@ -37,10 +37,7 @@
 from neo4j_graphrag.experimental.components.text_splitters.fixed_size_splitter import (
     FixedSizeSplitter,
 )
-from neo4j_graphrag.experimental.components.types import (
-    LexicalGraphConfig,
-    SchemaEnforcementMode
-)
+from neo4j_graphrag.experimental.components.types import LexicalGraphConfig, SchemaEnforcementMode
 from neo4j_graphrag.experimental.pipeline.config.object_config import ComponentType
 from neo4j_graphrag.experimental.pipeline.config.template_pipeline.base import (
     TemplatePipelineConfig,

tests/unit/experimental/components/test_entity_relation_extractor.py

@@ -29,9 +29,9 @@
 from neo4j_graphrag.experimental.components.types import (
     DocumentInfo,
     Neo4jGraph,
+    SchemaEnforcementMode,
     TextChunk,
     TextChunks,
-    SchemaEnforcementMode,
 )
 from neo4j_graphrag.experimental.pipeline.exceptions import InvalidJSONError
 from neo4j_graphrag.llm import LLMInterface, LLMResponse