From c3c36fba6c05ab9484aad74588e5e2865908527f Mon Sep 17 00:00:00 2001 From: Sheilah Date: Wed, 16 Feb 2022 16:28:01 -0800 Subject: [PATCH] added link to created bug-fix libcudf ticket --- python/cudf/cudf/core/column/column.py | 4 +++- python/cudf/cudf/tests/test_dataframe.py | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/python/cudf/cudf/core/column/column.py b/python/cudf/cudf/core/column/column.py index 5d5e85e4d79..1c1c2ef2bf6 100644 --- a/python/cudf/cudf/core/column/column.py +++ b/python/cudf/cudf/core/column/column.py @@ -335,7 +335,9 @@ def _fill( return self def shift(self, offset: int, fill_value: ScalarLike) -> ColumnBase: - # link to the libcudf ticket you will create + # libcudf currently doesn't handle case when offset > len(df) + # ticket to fix the bug in link below: + # https://github.com/rapidsai/cudf/issues/10314 if abs(offset) > len(self): if fill_value is None: return column_empty_like(self, masked=True) diff --git a/python/cudf/cudf/tests/test_dataframe.py b/python/cudf/cudf/tests/test_dataframe.py index 5b9c73fd827..f1468801732 100644 --- a/python/cudf/cudf/tests/test_dataframe.py +++ b/python/cudf/cudf/tests/test_dataframe.py @@ -9190,7 +9190,7 @@ def test_dataframe_rename_duplicate_column(): @pytest.mark.parametrize( "data", [ - np.random.RandomState(seed=10).randint(-50, 50, (25, 30)), + np.random.RandomState(seed=10).randint(-50, 50, (10, 10)), np.random.RandomState(seed=10).random_sample((4, 4)), np.array([1.123, 2.343, 5.890, 0.0]), {"a": [1.123, 2.343, np.nan, np.nan], "b": [None, 3, 9.08, None]},