Skip to content

Commit 9de05a4

Browse files
committed
revert minor changes
1 parent 5468a2e commit 9de05a4

2 files changed

Lines changed: 1 addition & 8 deletions

File tree

graphrag/storage/factory.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -40,9 +40,6 @@ def register(
4040
storage_type: The type identifier for the storage.
4141
creator: A class or callable that creates an instance of PipelineStorage.
4242
43-
Raises
44-
------
45-
TypeError: If creator is a class type instead of a factory function.
4643
"""
4744
cls._registry[storage_type] = creator
4845

graphrag/storage/memory_pipeline_storage.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,7 @@ class MemoryPipelineStorage(FilePipelineStorage):
1818

1919
def __init__(self):
2020
"""Init method definition."""
21-
# MemoryPipelineStorage doesn't need actual file storage, use temp dir
22-
import tempfile
23-
24-
temp_dir = tempfile.mkdtemp()
25-
super().__init__(base_dir=temp_dir)
21+
super().__init__()
2622
self._storage = {}
2723

2824
async def get(

0 commit comments

Comments
 (0)