Skip to content
4 changes: 2 additions & 2 deletions python/cudf/cudf/testing/_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,10 +72,10 @@
]


def set_random_null_mask_inplace(series, null_probability=0.5, seed=None):
def set_random_null_mask_inplace(series, null_probability=0.5):
"""Randomly nullify elements in series with the provided probability."""
probs = [null_probability, 1 - null_probability]
rng = np.random.default_rng(seed=seed)
rng = np.random.default_rng(seed=0)
mask = rng.choice([False, True], size=len(series), p=probs)
Comment thread
coderabbitai[bot] marked this conversation as resolved.
series.iloc[mask] = None

Expand Down
3 changes: 0 additions & 3 deletions python/cudf/cudf/tests/test_doctests.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,9 +97,6 @@ def _find_doctests_in_obj(obj, finder=None, criteria=None):
"register_series_accessor": pytest.mark.filterwarnings(
"ignore:Attribute .* will be overridden:UserWarning"
),
"StringMethods.edit_distance_matrix": pytest.mark.filterwarnings(
"ignore:edit_distance_matrix:FutureWarning"
),
}


Expand Down
Loading