diff --git a/crates/polars-compute/src/var_cov.rs b/crates/polars-compute/src/var_cov.rs index de65ad4362bd..6f932b679800 100644 --- a/crates/polars-compute/src/var_cov.rs +++ b/crates/polars-compute/src/var_cov.rs @@ -231,10 +231,11 @@ where } } -fn chunk_as_float_binary(it: I, mut f: F) +fn chunk_as_float_binary(it: I, mut f: F) where T: NativeType + AsPrimitive, - I: IntoIterator, + U: NativeType + AsPrimitive, + I: IntoIterator, F: FnMut(&[f64], &[f64]), { let mut left_chunk = [0.0; CHUNK_SIZE]; @@ -271,9 +272,10 @@ where out } -pub fn cov(x: &PrimitiveArray, y: &PrimitiveArray) -> CovState +pub fn cov(x: &PrimitiveArray, y: &PrimitiveArray) -> CovState where T: NativeType + AsPrimitive, + U: NativeType + AsPrimitive, { assert!(x.len() == y.len()); let mut out = CovState::default(); @@ -293,9 +295,10 @@ where out } -pub fn pearson_corr(x: &PrimitiveArray, y: &PrimitiveArray) -> PearsonState +pub fn pearson_corr(x: &PrimitiveArray, y: &PrimitiveArray) -> PearsonState where T: NativeType + AsPrimitive, + U: NativeType + AsPrimitive, { assert!(x.len() == y.len()); let mut out = PearsonState::default();