@@ -822,7 +822,6 @@ impl<'a> Pregel<'a> {
822822 // execution is performed until all the nodes vote to halt, or the number of iterations is
823823 // greater than the maximum number of iterations set by the user at the initialization of
824824 // the model (see the `Pregel::new` method). We start by setting the number of iterations to 1.
825- println ! ( "current_vertices: {:?}" , current_vertices) ;
826825 let mut iteration = 1 ;
827826 // TODO: check that nodes are not halted.
828827 while iteration <= self . max_iterations {
@@ -848,7 +847,6 @@ impl<'a> Pregel<'a> {
848847 Column :: edge ( Column :: Dst ) , // dst column of the resulting DataFrame
849848 Column :: dst ( Column :: Id ) , // id column of the current_vertices DataFrame
850849 ) ;
851- println ! ( "triplets_df: {:?}" , triplets_df. clone( ) . collect( ) ) ;
852850 // We create a DataFrame that contains the messages sent by the vertices. The messages
853851 // are computed by performing an aggregation on the `triplets_df` DataFrame. The aggregation
854852 // is performed on the `msg` column of the `triplets_df` DataFrame, and the aggregation
@@ -878,7 +876,6 @@ impl<'a> Pregel<'a> {
878876 . select ( send_messages)
879877 . groupby ( [ Column :: msg ( Some ( Column :: Id ) ) ] )
880878 . agg ( [ aggregate_messages ( ) . alias ( Column :: Pregel . as_ref ( ) ) ] ) ;
881- println ! ( "message_df: {:?}" , message_df. clone( ) . collect( ) ) ;
882879 // We Compute the new values for the vertices. Note that we have to check for possibly
883880 // null values after performing the outer join. This is, columns where the join key does
884881 // not exist in the source DataFrame. In case we find any; for example, given a certain
@@ -903,7 +900,6 @@ impl<'a> Pregel<'a> {
903900 col( Column :: Id . as_ref( ) ) ,
904901 v_prog( ) . alias( self . vertex_column. as_ref( ) ) ,
905902 ] ) ;
906- println ! ( "vertex_columns: {:?}" , vertex_columns. clone( ) . collect( ) ) ;
907903 // We update the `current_vertices` DataFrame with the new values for the vertices. We
908904 // do so by performing an inner join between the `current_vertices` DataFrame and the
909905 // `vertex_columns` DataFrame. The join is performed on the `id` column of the
@@ -985,10 +981,7 @@ mod tests {
985981 fn agg_pagerank ( pagerank : Pregel ) -> Result < f64 , String > {
986982 let result = match pagerank. run ( ) {
987983 Ok ( result) => result,
988- Err ( error) => {
989- println ! ( "{}" , error) ;
990- return Err ( String :: from ( "Error running the PageRank algorithm" ) ) ;
991- }
984+ Err ( _) => return Err ( String :: from ( "Error running the PageRank algorithm" ) ) ,
992985 } ;
993986 let rank = match result. column ( "rank" ) {
994987 Ok ( rank) => rank,
0 commit comments