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
5 changes: 3 additions & 2 deletions python/cudf_polars/tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -404,6 +404,7 @@ def pytest_collection_modifyitems(


@pytest.fixture(scope="module")
def parquet_stats_executor() -> concurrent.futures.ThreadPoolExecutor:
def parquet_stats_executor() -> concurrent.futures.ThreadPoolExecutor: # type: ignore[misc]
"""A thread pool to use for cudf-polars status collection."""
return concurrent.futures.ThreadPoolExecutor()
with concurrent.futures.ThreadPoolExecutor() as executor:
yield executor
Loading