Skip to content

Commit f9f48e9

Browse files
committed
fix: HALKWalker with reverse walking
1 parent 548ae55 commit f9f48e9

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pyrdf2vec/walkers/halk.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,10 @@ def _post_extract(self, res: List[EntityWalks]) -> List[List[SWalk]]:
147147
for i, vertex in enumerate(walk[1::2], 2):
148148
if vertex not in rare_predicates:
149149
obj = walk[i] if i % 2 == 0 else walk[i + 1]
150-
canonical_walk += [vertex, obj]
150+
if self.with_reverse:
151+
canonical_walk = [obj, vertex] + canonical_walk
152+
else:
153+
canonical_walk += [vertex, obj]
151154
if len(canonical_walk) >= 3:
152155
canonical_walks.append(tuple(canonical_walk))
153156
if canonical_walks:

0 commit comments

Comments
 (0)