apply suggestion

rapidsai · Aug 14, 2024 · 703b41a · 703b41a
1 parent e7d1bea
commit 703b41a
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/python/cuxfilter/charts/core/non_aggregate/core_graph.py b/python/cuxfilter/charts/core/non_aggregate/core_graph.py
@@ -50,14 +50,14 @@ def edge_columns(self):
 
     @property
     def node_columns(self):
+        cols = [
+            self.node_id,
+            self.node_x,
+            self.node_y,
+        ]
         if self.node_aggregate_col != self.node_id:
-            return [
-                self.node_id,
-                self.node_x,
-                self.node_y,
-                self.node_aggregate_col,
-            ]
-        return [self.node_id, self.node_x, self.node_y]
+            cols.append(self.node_aggregate_col)
+        return cols
 
     def __init__(
         self,