Skip to content

Commit 4908b6a

Browse files
committed
Fix mypy
1 parent f7a669a commit 4908b6a

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
@@ -691,7 +691,7 @@ def _extract_required_properties(
691691
existence_constraint.append((lab, prop))
692692
return existence_constraint
693693

694-
async def run(self, *args, **kwargs) -> GraphSchema:
694+
async def run(self, *args: Any, **kwargs: Any) -> GraphSchema:
695695
structured_schema = get_structured_schema(self.driver, database=self.database)
696696
existence_constraint = self._extract_required_properties(structured_schema)
697697

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)