Updating benchmarks test to reflect new UMAP benchmark name

albert17 · Apr 27, 2020 · 4b060f8 · 4b060f8
1 parent 7fdb3e7
commit 4b060f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cuml/test/test_benchmark.py b/python/cuml/test/test_benchmark.py
@@ -169,7 +169,7 @@ def predict(self, X):
 
 # Only test a few algorithms (which collectively span several types)
 # to reduce runtime burden
-@pytest.mark.parametrize('algo_name', ['UMAP',
+@pytest.mark.parametrize('algo_name', ['UMAP-Supervised',
                                        'DBSCAN',
                                        'LogisticRegression',
                                        'ElasticNet',