Remove 0 epsilon (#7883)

After discussion with @Linchenn, I found out that using a zero-value epsilon in expectTensorsClose causes some tests for devices with low precision to fail. This PR changes the tests to use the default epsilon value.
tensorflow · Jul 31, 2023 · 5974699 · 5974699
1 parent 4e7d5f6
commit 5974699
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tfjs-layers/src/layers/nlp/utils_test.ts b/tfjs-layers/src/layers/nlp/utils_test.ts
@@ -51,7 +51,7 @@ describe('sliceUpdate', () => {
 
     const result = sliceUpdate(inputs, startIndices, updates);
 
-    expectTensorsClose(result, expected, 0);
+    expectTensorsClose(result, expected);
   });
 
   it('2D', () => {
@@ -67,7 +67,7 @@ describe('sliceUpdate', () => {
     ]);
     const result = sliceUpdate(inputs, startIndices, updates);
 
-    expectTensorsClose(result, expected, 0);
+    expectTensorsClose(result, expected);
   });
 
   it('3D', () => {
@@ -84,6 +84,6 @@ describe('sliceUpdate', () => {
     ]);
     const result = sliceUpdate(inputs, startIndices, updates);
 
-    expectTensorsClose(result, expected, 0);
+    expectTensorsClose(result, expected);
   });
 });