Skip to content

Commit 857e20f

Browse files
committed
bug fix: keep list of composite keys within next_level dictionary
1 parent d3a6699 commit 857e20f

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

graphflex/neighbourhood/extractor.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,9 @@ def get_neighborhoods(self, noi, depth):
2727
composite_key = f"{key}¥{nkey}"
2828
else:
2929
composite_key = f"{nkey}"
30-
next_level[composite_key] = value
30+
if composite_key not in next_level:
31+
next_level[composite_key] = []
32+
next_level[composite_key].extend(value)
3133

3234
neighbours[d] = next_level
3335
return neighbours

0 commit comments

Comments
 (0)