From 562a281ce4726d24d6b8278cc72697c112e36c0a Mon Sep 17 00:00:00 2001 From: nitowa Date: Thu, 25 Aug 2022 13:48:05 -0400 Subject: [PATCH] rename variables in graph --- src/spark/main_graphs.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/spark/main_graphs.py b/src/spark/main_graphs.py index f715824..a4046c1 100644 --- a/src/spark/main_graphs.py +++ b/src/spark/main_graphs.py @@ -49,7 +49,7 @@ master.spark.sparkContext.setCheckpointDir('./checkpoints') # spark is really a tx_df = master.get_tx_dataframe() -transaction_as_vertices = tx_df \ +addresses_as_vertices = tx_df \ .select('address') \ .withColumnRenamed('address', 'id') \ .distinct() @@ -65,7 +65,7 @@ transactions_as_edges = tx_df \ .flatMap(explode_row) \ .toDF(['src', 'dst']) -g = GraphFrame(transaction_as_vertices, transactions_as_edges) +g = GraphFrame(addresses_as_vertices, transactions_as_edges) components = g.connectedComponents(algorithm='graphframes') master.write_connected_components_as_clusters(components)