[REVIEW] Fixing dask tsvd stress test failure (#2941)

* ADD int conversion * UPDATE changelog.md * ADD int to params Co-authored-by: Nanthini Balasubramanian <[email protected]>
rapidsai · Oct 15, 2020 · 5aa17ac · 5aa17ac
1 parent 0dc18e0
commit 5aa17ac
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -87,6 +87,7 @@
 - PR #2932: Marking KBinsDiscretizer pytests as xfail
 - PR #2925: Fixing Owner Bug When Slicing CumlArray Objects
 - PR #2931: Fix notebook error handling in gpuCI
+- PR #2941: Fixing dask tsvd stress test failure
 - PR #2943: Remove unused shuffle_features parameter
 - PR #2940: Correcting labels meta dtype for `cuml.dask.make_classification`
 - PR #2965: Notebooks update

diff --git a/python/cuml/test/dask/test_tsvd.py b/python/cuml/test/dask/test_tsvd.py
@@ -25,7 +25,7 @@
 
 @pytest.mark.mg
 @pytest.mark.parametrize("data_info", [unit_param([1000, 20, 30]),
-                         stress_param([9e6, 5000, 30])])
+                         stress_param([int(9e6), 5000, 30])])
 @pytest.mark.parametrize("input_type", ["dataframe", "array"])
 def test_pca_fit(data_info, input_type, client):
 
@@ -72,7 +72,7 @@ def test_pca_fit(data_info, input_type, client):
 
 @pytest.mark.mg
 @pytest.mark.parametrize("data_info", [unit_param([1000, 20, 46]),
-                         stress_param([9e6, 5000, 46])])
+                         stress_param([int(9e6), 5000, 46])])
 def test_pca_fit_transform_fp32(data_info, client):
 
     nrows, ncols, n_parts = data_info
@@ -92,7 +92,7 @@ def test_pca_fit_transform_fp32(data_info, client):
 
 @pytest.mark.mg
 @pytest.mark.parametrize("data_info", [unit_param([1000, 20, 33]),
-                         stress_param([9e6, 5000, 33])])
+                         stress_param([int(9e6), 5000, 33])])
 def test_pca_fit_transform_fp64(data_info, client):
 
     nrows, ncols, n_parts = data_info