From 02c634cf28a7b9913ba0e3495193be5926dafcd8 Mon Sep 17 00:00:00 2001 From: Praateek Mahajan Date: Tue, 24 Sep 2024 14:43:23 -0700 Subject: [PATCH] change how version is understood for cudf Signed-off-by: Praateek Mahajan --- crossfit/backend/cudf/series.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/crossfit/backend/cudf/series.py b/crossfit/backend/cudf/series.py index 44294e1..5e45d1f 100644 --- a/crossfit/backend/cudf/series.py +++ b/crossfit/backend/cudf/series.py @@ -28,10 +28,9 @@ @lru_cache def _is_cudf_gte_24_10(): - from importlib_metadata import version from packaging.version import parse as parse_version - current_cudf_version = parse_version(version("cudf_cu12")) + current_cudf_version = parse_version(cudf.__version__) cudf_24_10_version = parse_version("24.10.0") if current_cudf_version >= cudf_24_10_version or (