Skip to content

Commit 30c4258

Browse files
committed
clean up
Signed-off-by: Connor Tsui <connor.tsui20@gmail.com>
1 parent fcf3e67 commit 30c4258

1 file changed

Lines changed: 2 additions & 5 deletions

File tree

vortex-tensor/src/scalar_fns/cosine_similarity.rs

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -188,6 +188,7 @@ fn cosine_similarity_row<T: Float + NativePType>(a: &[T], b: &[T]) -> T {
188188
#[cfg(test)]
189189
mod tests {
190190
use rstest::rstest;
191+
use vortex::array::ArrayRef;
191192
use vortex::array::ToCanonical;
192193
use vortex::array::arrays::ScalarFnArray;
193194
use vortex::error::VortexResult;
@@ -202,11 +203,7 @@ mod tests {
202203
use crate::scalar_fns::utils::test_helpers::vector_array;
203204

204205
/// Evaluates cosine similarity between two tensor arrays and returns the result as `Vec<f64>`.
205-
fn eval_cosine_similarity(
206-
lhs: vortex::array::ArrayRef,
207-
rhs: vortex::array::ArrayRef,
208-
len: usize,
209-
) -> VortexResult<Vec<f64>> {
206+
fn eval_cosine_similarity(lhs: ArrayRef, rhs: ArrayRef, len: usize) -> VortexResult<Vec<f64>> {
210207
let scalar_fn = ScalarFn::new(CosineSimilarity, EmptyOptions).erased();
211208
let result = ScalarFnArray::try_new(scalar_fn, vec![lhs, rhs], len)?;
212209
let prim = result.to_primitive();

0 commit comments

Comments
 (0)