Ruff

Author: NathalieCharbel

examples/automatic_schema_extraction/schema_from_text.py

@@ -129,7 +129,7 @@ async def main() -> None:
     print("\nLoading schemas from saved files:")
     schema_from_json = SchemaConfig.from_file(JSON_FILE_PATH)
     schema_from_yaml = SchemaConfig.from_file(YAML_FILE_PATH)
-    
+
     print(f"Entities in JSON schema: {list(schema_from_json.entities.keys())}")
     print(f"Entities in YAML schema: {list(schema_from_yaml.entities.keys())}")
 

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

@@ -80,7 +80,9 @@ class SimpleKGPipelineConfig(TemplatePipelineConfig):
     entities: Sequence[EntityInputType] = []
     relations: Sequence[RelationInputType] = []
     potential_schema: Optional[list[tuple[str, str, str]]] = None
-    schema_: Optional[Union[SchemaConfig, dict[str, list[Any]]]] = Field(default=None, alias="schema")
+    schema_: Optional[Union[SchemaConfig, dict[str, list[Any]]]] = Field(
+        default=None, alias="schema"
+    )
     enforce_schema: SchemaEnforcementMode = SchemaEnforcementMode.NONE
     on_error: OnError = OnError.IGNORE
     prompt_template: Union[ERExtractionTemplate, str] = ERExtractionTemplate()
@@ -193,14 +195,16 @@ def _process_schema_with_precedence(
                 entity_dicts = list(self.schema_.entities.values())
                 # convert dict values to SchemaEntity objects
                 entities = [SchemaEntity.model_validate(e) for e in entity_dicts]
-                
+
                 # handle case where relations could be None
                 if self.schema_.relations is not None:
                     relation_dicts = list(self.schema_.relations.values())
-                    relations = [SchemaRelation.model_validate(r) for r in relation_dicts]
+                    relations = [
+                        SchemaRelation.model_validate(r) for r in relation_dicts
+                    ]
                 else:
                     relations = []
-                    
+
                 potential_schema = self.schema_.potential_schema
             else:
                 # extract from dictionary