From 8ad5d3bda6f29971e318e4f05241e2f0bf1890be Mon Sep 17 00:00:00 2001 From: Jinfeng Date: Mon, 7 Aug 2023 17:29:16 -0700 Subject: [PATCH] get pytests work for both cuml 23.08 and 23.06 Signed-off-by: Jinfeng --- python/tests/test_linear_model.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/python/tests/test_linear_model.py b/python/tests/test_linear_model.py index a1556dd5..f3c364f6 100644 --- a/python/tests/test_linear_model.py +++ b/python/tests/test_linear_model.py @@ -98,6 +98,12 @@ def test_default_cuml_params() -> None: [CumlLinearRegression, Ridge, CD], ["handle", "output_type"] ) spark_params = LinearRegression()._get_cuml_params_default() + + import cuml + from packaging import version + + if version.parse(cuml.__version__) < version.parse("23.08.00"): + spark_params.pop("copy_X") assert cuml_params == spark_params