From bed88234af0d588f670d2b475ba194c3b268bf2c Mon Sep 17 00:00:00 2001
From: lq <Qiang.D.Li@gmail.com>
Date: Fri, 8 May 2020 16:57:59 +0800
Subject: [PATCH] Fix: Return type of validate.calculate_similarity (#628)

When an input is zero, it returns an Int, which results to `TypeError` in `compare_output`
---
 tools/validate.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tools/validate.py b/tools/validate.py
index 8b8a9e82..4f780042 100644
--- a/tools/validate.py
+++ b/tools/validate.py
@@ -69,9 +69,9 @@ def calculate_similarity(u, v, data_type=np.float64):
     norm = u_norm * v_norm
     if norm == 0:
         if u_norm == 0 and v_norm == 0:
-            return 1
+            return data_type(1)
         else:
-            return 0
+            return data_type(0)
     else:
         return np.dot(u, v) / norm
 
-- 
GitLab