chore: replace print statements from logs (#1162)

Author: milenkovicm

ballista/core/src/consistent_hash/mod.rs

@@ -309,7 +309,6 @@ mod test {
 
         let keys = vec!["1", "4", "5", "3", "2"];
         for (i, key) in keys.iter().enumerate() {
-            // println!("{}", consistent_hash.get(key.as_bytes()).unwrap().name());
             assert_eq!(
                 consistent_hash.get(key.as_bytes()).unwrap().name(),
                 nodes[i].name()

ballista/core/src/diagram.rs

@@ -27,6 +27,7 @@ use datafusion::physical_plan::joins::HashJoinExec;
 use datafusion::physical_plan::projection::ProjectionExec;
 use datafusion::physical_plan::sorts::sort::SortExec;
 use datafusion::physical_plan::ExecutionPlan;
+use log::warn;
 use std::fs::File;
 use std::io::{BufWriter, Write};
 use std::sync::atomic::{AtomicUsize, Ordering};
@@ -110,7 +111,7 @@ fn build_exec_plan_diagram(
     {
         "CoalescePartitionsExec"
     } else {
-        println!("Unknown: {plan:?}");
+        warn!("Unknown: {plan:?}");
         "Unknown"
     };
 

ballista/scheduler/src/state/task_manager.rs

@@ -422,7 +422,7 @@ impl<T: 'static + AsLogicalPlan, U: 'static + AsExecutionPlan> TaskManager<T, U>
 
             graph.revive();
 
-            println!("Saving job with status {:?}", graph.status());
+            info!("Saving job with status {:?}", graph.status());
 
             self.state.save_job(job_id, &graph).await?;