Use common variable

rapidsai · Mar 4, 2022 · a7588b3 · a7588b3
1 parent 66acd59
commit a7588b3
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/python/cudf/cudf/tests/test_dataframe.py b/python/cudf/cudf/tests/test_dataframe.py
@@ -7240,10 +7240,9 @@ def test_sample_axis_0(
 def test_sample_reproducibility(replace, random_state_lib):
     df = cudf.DataFrame({"a": cupy.arange(0, 1024)})
 
-    expected = df.sample(
-        1024, replace=replace, random_state=random_state_lib(10)
-    )
-    out = df.sample(1024, replace=replace, random_state=random_state_lib(10))
+    n = 1024
+    expected = df.sample(n, replace=replace, random_state=random_state_lib(10))
+    out = df.sample(n, replace=replace, random_state=random_state_lib(10))
 
     assert_eq(expected, out)