defer compute for values parameter

rapidsai · Oct 11, 2022 · b413523 · b413523
1 parent cfda47b
commit b413523
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/python/cudf/benchmarks/API/bench_dataframe.py b/python/cudf/benchmarks/API/bench_dataframe.py
@@ -41,14 +41,14 @@ def bench_merge(benchmark, dataframe, num_key_cols):
 @pytest.mark.parametrize(
     "values",
     [
-        range(50),
-        {f"{string.ascii_lowercase[i]}": range(50) for i in range(10)},
-        cudf.DataFrame({f"{string.ascii_lowercase[i]}": range(50) for i in range(10)}),
-        cudf.Series(range(50)),
+        lambda: range(50),
+        lambda: {f"{string.ascii_lowercase[i]}": range(50) for i in range(10)},
+        lambda: cudf.DataFrame({f"{string.ascii_lowercase[i]}": range(50) for i in range(10)}),
+        lambda: cudf.Series(range(50)),
     ],
 )
 def bench_isin(benchmark, dataframe, values):
-    benchmark(dataframe.isin, values)
+    benchmark(dataframe.isin, values())
 
 
 @pytest.fixture(