get pytests work for both cuml 23.08 and 23.06

Signed-off-by: Jinfeng <[email protected]>
NVIDIA · Aug 8, 2023 · 8ad5d3b · 8ad5d3b
1 parent 0a10f5c
commit 8ad5d3b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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