From 7e2b042eb89daeb897811e0da5c246cf4d149f52 Mon Sep 17 00:00:00 2001 From: Paul Ganssle Date: Thu, 26 Jan 2017 11:30:30 -0500 Subject: [PATCH] Change rstrip() to truncation in test function --- sklearn/gaussian_process/tests/test_kernels.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sklearn/gaussian_process/tests/test_kernels.py b/sklearn/gaussian_process/tests/test_kernels.py index e670e53308892..003c5727da097 100644 --- a/sklearn/gaussian_process/tests/test_kernels.py +++ b/sklearn/gaussian_process/tests/test_kernels.py @@ -80,7 +80,7 @@ def test_kernel_theta(): # Determine kernel parameters that contribute to theta init_sign = signature(kernel.__class__.__init__).parameters.values() args = [p.name for p in init_sign if p.name != 'self'] - theta_vars = map(lambda s: s.rstrip("_bounds"), + theta_vars = map(lambda s: s[0:-len("_bounds")], filter(lambda s: s.endswith("_bounds"), args)) assert_equal( set(hyperparameter.name