File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
src/neo4j_graphrag/experimental/components
tests/unit/experimental/components Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -702,7 +702,7 @@ def _extract_required_properties(
702
702
existence_constraint .append ((lab , prop ))
703
703
return existence_constraint
704
704
705
- async def run (self , * args , ** kwargs ) -> GraphSchema :
705
+ async def run (self , * args : Any , ** kwargs : Any ) -> GraphSchema :
706
706
structured_schema = get_structured_schema (self .driver , database = self .database )
707
707
existence_constraint = self ._extract_required_properties (structured_schema )
708
708
Original file line number Diff line number Diff line change @@ -962,7 +962,7 @@ async def test_schema_from_text_filters_relationships_without_labels(
962
962
963
963
@pytest .mark .asyncio
964
964
@patch ("neo4j_graphrag.experimental.components.schema.get_structured_schema" )
965
- async def test_schema_from_existing_graph (mock_get_structured_schema : Mock ):
965
+ async def test_schema_from_existing_graph (mock_get_structured_schema : Mock ) -> None :
966
966
mock_get_structured_schema .return_value = {
967
967
"node_props" : {
968
968
"Person" : [
You can’t perform that action at this time.
0 commit comments