BUGFIX: fixing legacy use cases of get_successors (#21)

driskai · Aug 27, 2024 · 63fe025 · 63fe025
1 parent fa7b76a
commit 63fe025
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/drisk_api/graph_client.py b/python/drisk_api/graph_client.py
@@ -320,7 +320,7 @@ def get_view_axes_and_filters(
         x_node, y_node, *filters = [
             uuid
             for uuid, _ in sorted(
-                zip(*self.get_successors(view_node, weights=True)),
+                self.get_successors(view_node, weights=True),
                 key=lambda x: x[1],
             )
         ]
@@ -342,7 +342,7 @@ def add_nodes_to_view(
         x_node, y_node, *_ = [
             uuid
             for uuid, _ in sorted(
-                zip(*self.get_successors(view_node, weights=True)),
+                self.get_successors(view_node, weights=True),
                 key=lambda x: x[1],
             )
         ]