Skip to content

Commit 00fc875

Browse files
committed
Fix mypy
1 parent 15826ff commit 00fc875

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/neo4j_graphrag/experimental/components/schema.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -702,7 +702,7 @@ def _extract_required_properties(
702702
existence_constraint.append((lab, prop))
703703
return existence_constraint
704704

705-
async def run(self, *args, **kwargs) -> GraphSchema:
705+
async def run(self, *args: Any, **kwargs: Any) -> GraphSchema:
706706
structured_schema = get_structured_schema(self.driver, database=self.database)
707707
existence_constraint = self._extract_required_properties(structured_schema)
708708

tests/unit/experimental/components/test_schema.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -962,7 +962,7 @@ async def test_schema_from_text_filters_relationships_without_labels(
962962

963963
@pytest.mark.asyncio
964964
@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:
966966
mock_get_structured_schema.return_value = {
967967
"node_props": {
968968
"Person": [

0 commit comments

Comments
 (0)