Skip to content

Commit fb57c60

Browse files
Fix ruff error
1 parent 329d6df commit fb57c60

File tree

3 files changed

+7
-3
lines changed

3 files changed

+7
-3
lines changed

src/neo4j_graphrag/experimental/components/data_loader.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,4 +138,3 @@ async def run(
138138
document_type=DocumentType.MARKDOWN,
139139
),
140140
)
141-

tests/unit/experimental/components/test_data_loader.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,7 @@ def test_pdf_loader_backward_compat_reexport_module() -> None:
176176
DataLoader as DataLoaderDirect,
177177
PdfLoader as PdfLoaderDirect,
178178
)
179+
179180
with pytest.warns(DeprecationWarning, match="pdf_loader"):
180181
from neo4j_graphrag.experimental.components.pdf_loader import (
181182
DataLoader as DataLoaderReexport,

tests/unit/experimental/pipeline/config/template_pipeline/test_simple_kg_builder.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,11 +61,15 @@ def test_simple_kg_pipeline_config_file_loader_from_file_is_true() -> None:
6161

6262

6363
@pytest.mark.asyncio
64-
async def test_simple_kg_pipeline_config_default_file_loader_supports_markdown() -> None:
64+
async def test_simple_kg_pipeline_config_default_file_loader_supports_markdown() -> (
65+
None
66+
):
6567
config = SimpleKGPipelineConfig(from_file=True)
6668
loader = config._get_file_loader()
6769
assert loader is not None
68-
doc = await loader.run(filepath="tests/unit/experimental/components/sample_data/hello.md")
70+
doc = await loader.run(
71+
filepath="tests/unit/experimental/components/sample_data/hello.md"
72+
)
6973
assert doc.document_info.document_type == DocumentType.MARKDOWN
7074

7175

0 commit comments

Comments
 (0)