change how version is understood for cudf

Signed-off-by: Praateek Mahajan <[email protected]>
rapidsai · Sep 24, 2024 · 02c634c · 02c634c
1 parent 1e49f37
commit 02c634c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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 (