lint

Author: jayshrivastava

src/flight_service/do_get.rs

@@ -7,7 +7,8 @@ use crate::flight_service::trailing_flight_data_stream::TrailingFlightDataStream
 use crate::metrics::TaskMetricsCollector;
 use crate::metrics::proto::df_metrics_set_to_proto;
 use crate::protobuf::{
-    datafusion_error_to_tonic_status, stage_from_proto, AppMetadata, DistributedCodec, FlightAppMetadata, MetricsCollection, StageKey, TaskMetrics
+    AppMetadata, DistributedCodec, FlightAppMetadata, MetricsCollection, StageKey, TaskMetrics,
+    datafusion_error_to_tonic_status, stage_from_proto,
 };
 use arrow::array::RecordBatch;
 use arrow::datatypes::SchemaRef;

src/test_utils/plans.rs

@@ -6,7 +6,9 @@ use datafusion::{
 };
 
 use crate::{
-    execution_plans::{InputStage, NetworkCoalesceExec, NetworkShuffleExec}, protobuf::{stage_from_proto, StageKey}, StageExec
+    StageExec,
+    execution_plans::{InputStage, NetworkCoalesceExec, NetworkShuffleExec},
+    protobuf::StageKey,
 };
 
 /// count_plan_nodes counts the number of execution plan nodes in a plan using BFS traversal.
@@ -58,14 +60,16 @@ pub fn get_stages_and_stage_keys(
         }
 
         // Add any child stages
-        queue.extend(stage.input_stages_iter().map(|input_stage| {
-            match input_stage {
-                InputStage::Decoded(plan) => StageExec::from_dyn(plan),
-                InputStage::Encoded { .. } => {
-                    unimplemented!();    
-                }
-            }
-        }));
+        queue.extend(
+            stage
+                .input_stages_iter()
+                .map(|input_stage| match input_stage {
+                    InputStage::Decoded(plan) => StageExec::from_dyn(plan),
+                    InputStage::Encoded { .. } => {
+                        unimplemented!();
+                    }
+                }),
+        );
     }
     (stages_map, stage_keys)
 }