Fix flaky tests

hicder · Jan 4, 2025 · 9e9ccfe · 9e9ccfe
1 parent 57687ba
commit 9e9ccfe
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/rs/config/Cargo.toml b/rs/config/Cargo.toml
@@ -1,7 +1,7 @@
 [package]
 name = "config"
 version = "0.1.0"
-edition = "2024"
+edition = "2021"
 
 [dependencies]
 serde.workspace = true

diff --git a/rs/index/src/ivf/reader.rs b/rs/index/src/ivf/reader.rs
@@ -264,8 +264,8 @@ mod tests {
             let posting_list = transmute_u8_to_slice::<u64>(posting_list_byte_arr.unwrap());
 
             // It's possible that the posting list size is more than max_posting_list_size,
-            // but it should be less than 2.5x.
-            assert!(posting_list.len() <= 25);
+            // but it should be less than 3x.
+            assert!(posting_list.len() <= 30);
         }
     }
 }
diff --git a/rs/utils/src/distance/dot_product.rs b/rs/utils/src/distance/dot_product.rs
@@ -87,7 +87,7 @@ impl DistanceCalculator for DotProductDistanceCalculator {
             });
     }
 
-    #[inline(always)] 
+    #[inline(always)]
     fn outermost_op(x: f32) -> f32 {
         Self::neg_score(x)
     }
@@ -101,7 +101,7 @@ mod tests {
     fn test_dot_product_distance_calculator() {
         let a = generate_random_vector(128);
         let b = generate_random_vector(128);
-        let eps = 1e-5;
+        let eps = 2.0 * 1e-5;
         let result = DotProductDistanceCalculator::calculate(&a, &b);
         let expected = DotProductDistanceCalculator::calculate_scalar(&a, &b);
         assert!((result - expected).abs() < eps);