From 9c8f0cbce5ef0c1de9ffa362e6493cb3a7c5d216 Mon Sep 17 00:00:00 2001 From: Oliver Holworthy Date: Wed, 22 Mar 2023 11:50:36 +0000 Subject: [PATCH] Use None as default value of `cpu` in test_column_similarity --- tests/unit/ops/test_column_similarity.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/ops/test_column_similarity.py b/tests/unit/ops/test_column_similarity.py index 11b5cae383a..21761e79d7d 100644 --- a/tests/unit/ops/test_column_similarity.py +++ b/tests/unit/ops/test_column_similarity.py @@ -31,7 +31,7 @@ from nvtabular.ops.column_similarity import ColumnSimilarity -@pytest.mark.parametrize("cpu", [True, False]) +@pytest.mark.parametrize("cpu", [True, None]) @pytest.mark.parametrize("cpu_features", [True, False]) @pytest.mark.parametrize("on_device", [True, False]) @pytest.mark.parametrize("metric", ["tfidf", "cosine", "inner"])