More formatting

Author: sgrebnov

ballista/client/tests/context_checks.rs

@@ -24,15 +24,15 @@ mod supported {
         standalone_context_with_state,
     };
     use ballista_core::config::BallistaConfig;
-    use datafusion::arrow::array::StringArray;
     use datafusion::arrow::array::Array;
+    use datafusion::arrow::array::StringArray;
     use datafusion::arrow::record_batch::RecordBatch;
     use datafusion::physical_plan::collect;
     use datafusion::prelude::*;
     use datafusion::{assert_batches_eq, prelude::SessionContext};
     use rstest::*;
-    use std::sync::Arc;
     use std::path::PathBuf;
+    use std::sync::Arc;
 
     #[rstest::fixture]
     fn test_data() -> String {

ballista/core/src/serde/mod.rs

@@ -852,8 +852,7 @@ mod test {
             plan_message.try_encode(&mut buf).unwrap();
 
             let decoded_message = LogicalPlanNode::try_decode(&buf).unwrap();
-            let decoded =
-                decoded_message.try_into_logical_plan(&ctx, &codec).unwrap();
+            let decoded = decoded_message.try_into_logical_plan(&ctx, &codec).unwrap();
 
             let LogicalPlan::Extension(ext) = &decoded else {
                 panic!("expected Extension, got {decoded:?}");
@@ -889,8 +888,8 @@ mod test {
             ExplainFormat::Graphviz,
         ] {
             let s = BallistaExplainNode::format_as_str(&f);
-            let parsed = BallistaExplainNode::format_from_str(s)
-                .expect("round-trip must succeed");
+            let parsed =
+                BallistaExplainNode::format_from_str(s).expect("round-trip must succeed");
             assert_eq!(parsed, f);
         }
         assert!(BallistaExplainNode::format_from_str("bogus").is_none());

ballista/executor/src/execution_loop.rs

@@ -520,9 +520,9 @@ mod tests {
         CreateUpdateSessionParams, CreateUpdateSessionResult, ExecuteQueryParams,
         ExecuteQueryResult, ExecutorRegistration, ExecutorStoppedParams,
         ExecutorStoppedResult, GetCatalogParams, GetCatalogResult, GetJobMetricsParams,
-        GetJobMetricsResult, GetJobStatusParams,
-        GetJobStatusResult, GetRemoteFunctionsParams, GetRemoteFunctionsResult,
-        HeartBeatParams, HeartBeatResult, RegisterExecutorParams, RegisterExecutorResult,
+        GetJobMetricsResult, GetJobStatusParams, GetJobStatusResult,
+        GetRemoteFunctionsParams, GetRemoteFunctionsResult, HeartBeatParams,
+        HeartBeatResult, RegisterExecutorParams, RegisterExecutorResult,
         RemoveSessionParams, RemoveSessionResult, UpdateTaskStatusParams,
         UpdateTaskStatusResult,
     };

ballista/scheduler/src/scheduler_server/grpc.rs

@@ -27,12 +27,12 @@ use ballista_core::serde::protobuf::{
     ExecuteQueryFailureResult, ExecuteQueryParams, ExecuteQueryResult,
     ExecuteQuerySuccessResult, ExecutorHeartbeat, ExecutorStoppedParams,
     ExecutorStoppedResult, GetCatalogParams, GetCatalogResult, GetJobMetricsParams,
-    GetJobMetricsResult, GetJobStatusParams,
-    GetJobStatusResult, GetRemoteFunctionsParams, GetRemoteFunctionsResult,
-    HeartBeatParams, HeartBeatResult, JobStatus, KeyValuePair, PollWorkParams,
-    PollWorkResult, RegisterExecutorParams, RegisterExecutorResult, RemoveSessionParams,
-    RemoveSessionResult, UpdateTaskStatusParams, UpdateTaskStatusResult,
-    execute_query_failure_result, execute_query_result,
+    GetJobMetricsResult, GetJobStatusParams, GetJobStatusResult,
+    GetRemoteFunctionsParams, GetRemoteFunctionsResult, HeartBeatParams, HeartBeatResult,
+    JobStatus, KeyValuePair, PollWorkParams, PollWorkResult, RegisterExecutorParams,
+    RegisterExecutorResult, RemoveSessionParams, RemoveSessionResult,
+    UpdateTaskStatusParams, UpdateTaskStatusResult, execute_query_failure_result,
+    execute_query_result,
 };
 use ballista_core::serde::scheduler::ExecutorMetadata;
 use datafusion_proto::logical_plan::AsLogicalPlan;
@@ -783,18 +783,17 @@ fn serialize_stage_metrics(
         };
 
         let metrics = if plan.metrics().is_some() {
-            let metrics: ballista_core::serde::protobuf::OperatorMetricsSet =
-                raw_metrics
-                    .get(metric_index)
-                    .ok_or_else(|| {
-                        BallistaError::Internal(format!(
-                            "Missing metrics for operator {} at depth {}",
-                            plan.name(),
-                            depth
-                        ))
-                    })?
-                    .clone()
-                    .try_into()?;
+            let metrics: ballista_core::serde::protobuf::OperatorMetricsSet = raw_metrics
+                .get(metric_index)
+                .ok_or_else(|| {
+                    BallistaError::Internal(format!(
+                        "Missing metrics for operator {} at depth {}",
+                        plan.name(),
+                        depth
+                    ))
+                })?
+                .clone()
+                .try_into()?;
             metric_index += 1;
             metrics.metrics
         } else {