From 430f642394b14c81410f51fc51519b96e21478ec Mon Sep 17 00:00:00 2001
From: Guillaume Lagrange <lagrange.guillaume.1@gmail.com>
Date: Tue, 19 Mar 2024 12:26:21 -0400
Subject: [PATCH] Change assert_approx_eq precision from 3 to 2 (#1491)

---
 crates/burn-jit/src/tests/reduce.rs | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/crates/burn-jit/src/tests/reduce.rs b/crates/burn-jit/src/tests/reduce.rs
index 24225889a..6987a86d6 100644
--- a/crates/burn-jit/src/tests/reduce.rs
+++ b/crates/burn-jit/src/tests/reduce.rs
@@ -124,7 +124,7 @@ mod reduction {
         ));
         let val_ref = tensor_ref.sum_dim(reduce_dim);
 
-        val_ref.into_data().assert_approx_eq(&val.into_data(), 3);
+        val_ref.into_data().assert_approx_eq(&val.into_data(), 2);
     }
 
     #[test]
@@ -142,7 +142,7 @@ mod reduction {
         ));
         let val_ref = tensor_ref.sum_dim(reduce_dim);
 
-        val_ref.into_data().assert_approx_eq(&val.into_data(), 3);
+        val_ref.into_data().assert_approx_eq(&val.into_data(), 2);
     }
 
     #[test]
@@ -160,7 +160,7 @@ mod reduction {
         ));
         let val_ref = tensor_ref.sum_dim(reduce_dim);
 
-        val_ref.into_data().assert_approx_eq(&val.into_data(), 3);
+        val_ref.into_data().assert_approx_eq(&val.into_data(), 2);
     }
 
     #[test]
@@ -181,7 +181,7 @@ mod reduction {
         ));
         let val_ref = tensor_ref.sum_dim(reduce_dim);
 
-        val_ref.into_data().assert_approx_eq(&val.into_data(), 3);
+        val_ref.into_data().assert_approx_eq(&val.into_data(), 2);
     }
 
     #[test]
@@ -199,7 +199,7 @@ mod reduction {
         ));
         let val_ref = tensor_ref.mean_dim(reduce_dim);
 
-        val_ref.into_data().assert_approx_eq(&val.into_data(), 3);
+        val_ref.into_data().assert_approx_eq(&val.into_data(), 2);
     }
 
     #[test]
@@ -254,7 +254,7 @@ mod reduction {
         ));
         let val_ref = tensor_ref.sum();
 
-        val_ref.into_data().assert_approx_eq(&val.into_data(), 3);
+        val_ref.into_data().assert_approx_eq(&val.into_data(), 2);
     }
 
     #[test]
@@ -270,7 +270,7 @@ mod reduction {
         ));
         let val_ref = tensor_ref.prod();
 
-        val_ref.into_data().assert_approx_eq(&val.into_data(), 3);
+        val_ref.into_data().assert_approx_eq(&val.into_data(), 2);
     }
 
     #[test]