From 85b07f69f7e49e41926a6c38a3679e90835d46f2 Mon Sep 17 00:00:00 2001 From: Lawrence Mitchell Date: Wed, 13 Mar 2024 15:33:03 +0000 Subject: [PATCH] Reintroduce PANDAS_GE_220 import This was required by #15109, but removed by the changes in #15145 and the merge order was such that they weren't tested against each other. --- python/cudf/cudf/tests/test_groupby.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/cudf/cudf/tests/test_groupby.py b/python/cudf/cudf/tests/test_groupby.py index befa9b467dd..06516b6b4ea 100644 --- a/python/cudf/cudf/tests/test_groupby.py +++ b/python/cudf/cudf/tests/test_groupby.py @@ -20,7 +20,11 @@ import cudf from cudf import DataFrame, Series from cudf.api.extensions import no_default -from cudf.core._compat import PANDAS_CURRENT_SUPPORTED_VERSION, PANDAS_VERSION +from cudf.core._compat import ( + PANDAS_CURRENT_SUPPORTED_VERSION, + PANDAS_GE_220, + PANDAS_VERSION, +) from cudf.core.udf._ops import arith_ops, comparison_ops, unary_ops from cudf.core.udf.groupby_typing import SUPPORTED_GROUPBY_NUMPY_TYPES from cudf.core.udf.utils import UDFError, precompiled