Add error tolerance for tensor equals

1 files changed, 3 insertions, 1 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java b/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java
index 5e3af70cba4..5ef23e34198 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java
@@ -259,7 +259,9 @@ public interface Tensor {
         if ( a.size() != b.size()) return false;
         for (Iterator<Cell> aIterator = a.cellIterator(); aIterator.hasNext(); ) {
             Cell aCell = aIterator.next();
-            if ( ! aCell.getValue().equals(b.get(aCell.getKey()))) return false;
+            double aValue = aCell.getValue();
+            double bValue = b.get(aCell.getKey());
+            if (Math.abs(aValue-bValue) > 1e-7) return false; // TODO: determine relative precision
         }
         return true;
     }