diff --git a/src/daft-micropartition/src/ops/agg.rs b/src/daft-micropartition/src/ops/agg.rs index 290a49197e..d9487fc04a 100644 --- a/src/daft-micropartition/src/ops/agg.rs +++ b/src/daft-micropartition/src/ops/agg.rs @@ -7,7 +7,7 @@ use crate::micropartition::MicroPartition; impl MicroPartition { pub fn agg(&self, to_agg: &[Expr], group_by: &[Expr]) -> DaftResult { - let io_stats = IOStatsContext::new(format!("MicroPartition::agg")); + let io_stats = IOStatsContext::new("MicroPartition::agg".to_string()); let tables = self.concat_or_get(io_stats)?; diff --git a/src/daft-micropartition/src/ops/concat.rs b/src/daft-micropartition/src/ops/concat.rs index 1a2112eb90..7190aac104 100644 --- a/src/daft-micropartition/src/ops/concat.rs +++ b/src/daft-micropartition/src/ops/concat.rs @@ -27,7 +27,7 @@ impl MicroPartition { } } - let io_stats = IOStatsContext::new(format!("MicroPartition::concat")); + let io_stats = IOStatsContext::new("MicroPartition::concat".to_string()); let mut all_tables = vec![]; diff --git a/src/daft-micropartition/src/ops/eval_expressions.rs b/src/daft-micropartition/src/ops/eval_expressions.rs index 1e81d05f4d..0274bbebef 100644 --- a/src/daft-micropartition/src/ops/eval_expressions.rs +++ b/src/daft-micropartition/src/ops/eval_expressions.rs @@ -31,7 +31,7 @@ fn infer_schema(exprs: &[Expr], schema: &Schema) -> DaftResult { impl MicroPartition { pub fn eval_expression_list(&self, exprs: &[Expr]) -> DaftResult { - let io_stats = IOStatsContext::new(format!("MicroPartition::eval_expression_list")); + let io_stats = IOStatsContext::new("MicroPartition::eval_expression_list".to_string()); let expected_schema = infer_schema(exprs, &self.schema)?; let tables = self.tables_or_read(io_stats)?; @@ -54,7 +54,7 @@ impl MicroPartition { } pub fn explode(&self, exprs: &[Expr]) -> DaftResult { - let io_stats = IOStatsContext::new(format!("MicroPartition::explode")); + let io_stats = IOStatsContext::new("MicroPartition::explode".to_string()); let tables = self.tables_or_read(io_stats)?; let evaluated_tables = tables diff --git a/src/daft-micropartition/src/ops/filter.rs b/src/daft-micropartition/src/ops/filter.rs index 56f5acf7e7..ee1d5a4e63 100644 --- a/src/daft-micropartition/src/ops/filter.rs +++ b/src/daft-micropartition/src/ops/filter.rs @@ -9,7 +9,7 @@ use daft_stats::TruthValue; impl MicroPartition { pub fn filter(&self, predicate: &[Expr]) -> DaftResult { - let io_stats = IOStatsContext::new(format!("MicroPartition::filter")); + let io_stats = IOStatsContext::new("MicroPartition::filter".to_string()); if predicate.is_empty() { return Ok(Self::empty(Some(self.schema.clone()))); } diff --git a/src/daft-micropartition/src/ops/join.rs b/src/daft-micropartition/src/ops/join.rs index 49fce45e52..c6d7b94537 100644 --- a/src/daft-micropartition/src/ops/join.rs +++ b/src/daft-micropartition/src/ops/join.rs @@ -34,7 +34,7 @@ impl MicroPartition { if let TruthValue::False = tv { return Ok(Self::empty(Some(join_schema.into()))); } - let io_stats = IOStatsContext::new(format!("MicroPartition::join")); + let io_stats = IOStatsContext::new("MicroPartition::join".to_string()); let lt = self.concat_or_get(io_stats.clone())?; let rt = right.concat_or_get(io_stats)?; diff --git a/src/daft-micropartition/src/ops/partition.rs b/src/daft-micropartition/src/ops/partition.rs index 9d07343304..e49d275610 100644 --- a/src/daft-micropartition/src/ops/partition.rs +++ b/src/daft-micropartition/src/ops/partition.rs @@ -46,7 +46,7 @@ impl MicroPartition { exprs: &[Expr], num_partitions: usize, ) -> DaftResult> { - let io_stats = IOStatsContext::new(format!("MicroPartition::partition_by_hash")); + let io_stats = IOStatsContext::new("MicroPartition::partition_by_hash".to_string()); let tables = self.tables_or_read(io_stats)?; @@ -66,7 +66,7 @@ impl MicroPartition { } pub fn partition_by_random(&self, num_partitions: usize, seed: u64) -> DaftResult> { - let io_stats = IOStatsContext::new(format!("MicroPartition::partition_by_random")); + let io_stats = IOStatsContext::new("MicroPartition::partition_by_random".to_string()); let tables = self.tables_or_read(io_stats)?; @@ -92,7 +92,7 @@ impl MicroPartition { boundaries: &Table, descending: &[bool], ) -> DaftResult> { - let io_stats = IOStatsContext::new(format!("MicroPartition::partition_by_range")); + let io_stats = IOStatsContext::new("MicroPartition::partition_by_range".to_string()); let tables = self.tables_or_read(io_stats)?; diff --git a/src/daft-micropartition/src/ops/sort.rs b/src/daft-micropartition/src/ops/sort.rs index 476e00674c..da26c7940a 100644 --- a/src/daft-micropartition/src/ops/sort.rs +++ b/src/daft-micropartition/src/ops/sort.rs @@ -10,7 +10,7 @@ use crate::micropartition::MicroPartition; impl MicroPartition { pub fn sort(&self, sort_keys: &[Expr], descending: &[bool]) -> DaftResult { - let io_stats = IOStatsContext::new(format!("MicroPartition::sort")); + let io_stats = IOStatsContext::new("MicroPartition::sort".to_string()); let tables = self.concat_or_get(io_stats)?; match tables.as_slice() { @@ -28,7 +28,7 @@ impl MicroPartition { } pub fn argsort(&self, sort_keys: &[Expr], descending: &[bool]) -> DaftResult { - let io_stats = IOStatsContext::new(format!("MicroPartition::argsort")); + let io_stats = IOStatsContext::new("MicroPartition::argsort".to_string()); let tables = self.concat_or_get(io_stats)?; match tables.as_slice() { diff --git a/src/daft-micropartition/src/ops/take.rs b/src/daft-micropartition/src/ops/take.rs index ffa190c163..f1e988ed36 100644 --- a/src/daft-micropartition/src/ops/take.rs +++ b/src/daft-micropartition/src/ops/take.rs @@ -9,7 +9,7 @@ use crate::micropartition::MicroPartition; impl MicroPartition { pub fn take(&self, idx: &Series) -> DaftResult { - let io_stats = IOStatsContext::new(format!("MicroPartition::take")); + let io_stats = IOStatsContext::new("MicroPartition::take".to_string()); let tables = self.concat_or_get(io_stats)?; match tables.as_slice() {