ci: fix ci error

jiangyi15 · Oct 25, 2024 · 4e1121f · 4e1121f
1 parent 2df461d
commit 4e1121f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tf_pwa/adaptive_bins.py b/tf_pwa/adaptive_bins.py
@@ -81,7 +81,8 @@ def single_split_bound(data, n=2, base_bound=None):
         """split data in the order of data value
 
         >>> data = np.array([1.0, 2.0, 1.4, 3.1])
-        >>> AdaptiveBound.single_split_bound(data)
+        >>> bound = AdaptiveBound.single_split_bound(data)
+        >>> [(float(i[0]+1e-6), float(i[1]+1e-6)) for i in bound]
         [(1.0, 1.7...), (1.7..., 3.1...)]
 
         """
@@ -102,7 +103,7 @@ def multi_split_bound(datas, n, base_bound=None):
 
         >>> data = np.array([[1.0, 2.0, 1.4, 3.1], [2.0, 1.0, 3.0, 1.0]])
         >>> bound, _ = AdaptiveBound.multi_split_bound(data, [2, 1])
-        >>> [(i[0][0]+1e-6, i[1][0]+1e-6) for i in bound]
+        >>> [(float(i[0][0]+1e-6), float(i[1][0]+1e-6)) for i in bound]
         [(1.0..., 1.7...), (1.7..., 3.1...)]
 
         """