Skip to content

Commit

Permalink
[REVIEW] Reducing dask coordinate descent test runtime (#3074)
Browse files Browse the repository at this point in the history
* Remove test

* UPDATE changelog

Co-authored-by: Nanthini Balasubramanian <nathanb@nvidia.com>
  • Loading branch information
Nanthini10 and Nanthini10 authored Oct 29, 2020
1 parent 5fff222 commit 7c6b142
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
- PR #2996: Removing the max_depth restriction for switching to the batched backend
- PR #3004: Remove Single Process Multi GPU (SPMG) code
- PR #3044: Move leftover `linalg` and `stats` to RAFT namespaces
- PR #3074: Reducing dask coordinate descent test runtime

## Bug Fixes
- PR #3072: Fusing metrics and score directories in src_prims
Expand Down
8 changes: 4 additions & 4 deletions python/cuml/test/dask/test_coordinate_descent.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,9 +26,9 @@

@pytest.mark.mg
@pytest.mark.parametrize('dtype', [np.float32, np.float64])
@pytest.mark.parametrize('alpha', [0.1, 0.001])
@pytest.mark.parametrize('alpha', [0.001])
@pytest.mark.parametrize('algorithm', ['cyclic', 'random'])
@pytest.mark.parametrize('nrows', [unit_param(500),
@pytest.mark.parametrize('nrows', [unit_param(50),
quality_param(5000),
stress_param(500000)])
@pytest.mark.parametrize('column_info', [unit_param([20, 10]),
Expand Down Expand Up @@ -63,7 +63,7 @@ def test_lasso(dtype, alpha, algorithm,

@pytest.mark.mg
@pytest.mark.parametrize('dtype', [np.float32, np.float64])
@pytest.mark.parametrize('nrows', [unit_param(500),
@pytest.mark.parametrize('nrows', [unit_param(50),
quality_param(5000),
stress_param(500000)])
@pytest.mark.parametrize('column_info', [unit_param([20, 10]),
Expand Down Expand Up @@ -92,7 +92,7 @@ def test_lasso_default(dtype, nrows, column_info, n_parts, client):


@pytest.mark.parametrize('dtype', [np.float32, np.float64])
@pytest.mark.parametrize('alpha', [0.2, 0.7])
@pytest.mark.parametrize('alpha', [0.5])
@pytest.mark.parametrize('algorithm', ['cyclic', 'random'])
@pytest.mark.parametrize('nrows', [unit_param(500),
quality_param(5000),
Expand Down

0 comments on commit 7c6b142

Please sign in to comment.