get pytests work for both cuml 23.08 and 23.06

NVIDIA · Aug 8, 2023 · cb213f2 · cb213f2
1 parent 6605ee1
commit cb213f2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/python/tests/test_linear_model.py b/python/tests/test_linear_model.py
@@ -98,6 +98,11 @@ 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