Skip to content

Commit 91211b9

Browse files
Corrected issue with returning edge model instances from node methods
1 parent fd6eed1 commit 91211b9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

django_postgresql_dag/models.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -248,7 +248,7 @@ def descendants_edges(self):
248248
"""
249249
Returns a queryset of descendants edges
250250
"""
251-
return self.filter_order_ids(self.descendants_edges_ids())
251+
return _filter_order(edge_model.objects, "pk", self.descendants_edges_ids())
252252

253253
def ancestors_edges_ids(self, cached_results=None):
254254
"""
@@ -271,7 +271,7 @@ def ancestors_edges(self):
271271
"""
272272
Returns a queryset of ancestors edges
273273
"""
274-
return self.filter_order_ids(self.ancestors_edges_ids())
274+
return _filter_order(edge_model.objects, "pk", self.ancestors_edges_ids())
275275

276276
def clan_edges_ids(self):
277277
"""
@@ -286,7 +286,7 @@ def clan_edges(self):
286286
"""
287287
Returns a queryset of all edges associated with a given node
288288
"""
289-
return self.filter_order_ids(self.clan_edges_ids())
289+
return _filter_order(edge_model.objects, "pk", self.clan_edges_ids())
290290

291291
def path_ids_list(
292292
self, target_node, directional=True, max_depth=20, max_paths=1

0 commit comments

Comments
 (0)