From 4e1121f448bd1f05eefb521cb56697edadfd7a98 Mon Sep 17 00:00:00 2001 From: jiangyi15 Date: Fri, 25 Oct 2024 22:56:16 +0800 Subject: [PATCH] ci: fix ci error --- tf_pwa/adaptive_bins.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tf_pwa/adaptive_bins.py b/tf_pwa/adaptive_bins.py index 192257a5..d91643ea 100644 --- 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...)] """