diff --git a/generate_hnet_training_data.py b/generate_hnet_training_data.py
index 774f63b..7a9929a 100644
--- a/generate_hnet_training_data.py
+++ b/generate_hnet_training_data.py
@@ -2,7 +2,7 @@
 import random
 import pickle
 from IPython import  embed
-eps = np.finfo(np.float).eps
+eps = np.finfo(np.float64).eps
 from scipy.optimize import linear_sum_assignment
 from scipy.spatial import distance