Skip to content

Commit 2198c01

Browse files
committed
fixed comparison of node in crawler
fixed bug in cloning edge as it was adding it twice to the node
1 parent e3f902d commit 2198c01

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

model_extraction/crawler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -380,7 +380,7 @@ def explore(self, current_node=None, current_edge=None):
380380
already_exists = False
381381
if actual_node:
382382
for other_edge in edge.tail.edges:
383-
if other_edge.head == actual_node:
383+
if other_edge.head is actual_node:
384384
already_exists = True
385385
current_path.append(other_edge)
386386
break

model_extraction/graph.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,6 @@ def clone(self):
150150
candidate += 1
151151
new_one = self.tail.create_edge(candidate_name, self.head)
152152
new_one.return_to_caller = self.return_to_caller
153-
new_one.tail.edges.append(new_one)
154153
return new_one
155154

156155

0 commit comments

Comments
 (0)