Add comment to specify number of digits of precision

llvm · Jun 8, 2022 · 28c50e0 · 28c50e0
1 parent f75c473
commit 28c50e0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/e2e_testing/lazy_tensor_core/main.py b/e2e_testing/lazy_tensor_core/main.py
@@ -24,6 +24,7 @@ class LTCNumericTests(unittest.TestCase):
     def assert_tensors_almost_equal(self, tensor_a, tensor_b, message):
         a, b = tensor_a.cpu().detach().numpy(), tensor_b.cpu().detach().numpy()
 
+        # Ensure tensors match up to 7 decimals of precision.
         assert_almost_equal(a, b, 7, message)
 
     def run_test(self, run_model):