Skip to content

Commit

Permalink
minor typo fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
jgbradley1 committed Dec 16, 2024
1 parent 09037b6 commit b959e2b
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 4 deletions.
3 changes: 0 additions & 3 deletions graphrag/api/query.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,9 +49,6 @@
from graphrag.vector_stores.base import BaseVectorStore
from graphrag.vector_stores.factory import VectorStoreFactory

if TYPE_CHECKING:
from graphrag.query.structured_search.base import SearchResult

if TYPE_CHECKING:
from graphrag.query.structured_search.base import SearchResult

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,7 @@ async def summarize_communities(
reports: list[CommunityReport | None] = []
tick = progress_ticker(callbacks.progress, len(local_contexts))
runner = load_strategy(strategy["type"])

for level in levels:
level_contexts = prep_community_report_context(
pd.DataFrame(reports),
Expand Down
2 changes: 1 addition & 1 deletion graphrag/storage/factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ class StorageFactory:

@classmethod
def register(cls, storage_type: str, storage: type):
"""Register a vector store type."""
"""Register a custom storage implementation."""
cls.storage_types[storage_type] = storage

@classmethod
Expand Down

0 comments on commit b959e2b

Please sign in to comment.