diff --git a/python/cudf_polars/tests/expressions/test_rolling.py b/python/cudf_polars/tests/expressions/test_rolling.py index d6e202b2e11b..831ee3dabeb0 100644 --- a/python/cudf_polars/tests/expressions/test_rolling.py +++ b/python/cudf_polars/tests/expressions/test_rolling.py @@ -358,6 +358,7 @@ def test_rank_over_with_null_values( @pytest.mark.parametrize("method", ["ordinal", "dense", "min", "max", "average"]) @pytest.mark.parametrize("descending", [False, True]) @pytest.mark.parametrize("order_by", [None, ["g2", pl.col("x2") * 2]]) +@pytest.mark.timeout(120) def test_rank_over_with_null_group_keys( engine: pl.GPUEngine, df: pl.LazyFrame, diff --git a/python/cudf_polars/tests/streaming/test_sort.py b/python/cudf_polars/tests/streaming/test_sort.py index 533b5933a025..4bdb56de5829 100644 --- a/python/cudf_polars/tests/streaming/test_sort.py +++ b/python/cudf_polars/tests/streaming/test_sort.py @@ -87,6 +87,7 @@ def large_frames(): ) +@pytest.mark.timeout(120) def test_sort(df, engine): q = df.sort(by=["y", "z"]) assert_gpu_result_equal(q, engine=engine)