Skip to content

Commit 5202d73

Browse files
Fix storage implementations to declare parent fields
Co-authored-by: ChenZiHong-Gavin <[email protected]>
1 parent 35fbbfe commit 5202d73

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

graphgen/models/storage/json_storage.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@
77

88
@dataclass
99
class JsonKVStorage(BaseKVStorage):
10+
working_dir: str = None
11+
namespace: str = None
1012
_data: dict[str, str] = None
1113

1214
def __post_init__(self):
@@ -53,6 +55,8 @@ async def drop(self):
5355

5456
@dataclass
5557
class JsonListStorage(BaseListStorage):
58+
working_dir: str = None
59+
namespace: str = None
5660
_data: list = None
5761

5862
def __post_init__(self):

graphgen/models/storage/networkx_storage.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,9 @@
1111

1212
@dataclass
1313
class NetworkXStorage(BaseGraphStorage):
14+
working_dir: str = None
15+
namespace: str = None
16+
1417
@staticmethod
1518
def load_nx_graph(file_name) -> Optional[nx.Graph]:
1619
if os.path.exists(file_name):

0 commit comments

Comments
 (0)