Skip to content

Commit 14e2892

Browse files
author
Kavyansh Chourasia
committed
Shiksha Ingestion: Added Knowledge graph steps to __init__
1 parent 80b285f commit 14e2892

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

shiksha-ingestion/pipeline_steps/__init__.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,12 @@
1010
from .step_5_subtopic_chapter_level_los_extraction import (
1111
SubtopicChapterLOsExtractionStep,
1212
)
13+
from .knowledge_graph import (
14+
ExtractChapterEntitiesStep,
15+
ExtractEntityContentStep,
16+
ExtractEntityRelationshipsStep,
17+
ConstructKnowledgeGraphStep,
18+
)
1319
from .step_6_subtopic_extraction_rule_based_cleaning import SubtopicCleaningStep
1420
from .step_7_subtopic_wise_lo_extraction import SubtopicWiseLOExtractionStep
1521
from .step_8_create_indexes import CreateIndexStep
@@ -24,6 +30,10 @@
2430
# "TextExtractionOLMOcrStep",
2531
"TextCleaningStep",
2632
"SubtopicChapterLOsExtractionStep",
33+
"ExtractChapterEntitiesStep",
34+
"ExtractEntityContentStep",
35+
"ExtractEntityRelationshipsStep",
36+
"ConstructKnowledgeGraphStep",
2737
"SubtopicCleaningStep",
2838
"SubtopicWiseLOExtractionStep",
2939
"CreateIndexStep",

0 commit comments

Comments
 (0)