remove WheelHelpers, add cudf import

rapidsai · Aug 29, 2024 · fe7df5d · fe7df5d
1 parent a5b6e4e
commit fe7df5d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 65 deletions.
diff --git a/python/cuspatial/CMakeLists.txt b/python/cuspatial/CMakeLists.txt
@@ -43,11 +43,7 @@ if(NOT cuspatial_FOUND)
   add_subdirectory(../../cpp cuspatial-cpp EXCLUDE_FROM_ALL)
 
   set(cython_lib_dir cuspatial)
-  include(cmake/Modules/WheelHelpers.cmake)
-  install_aliased_imported_targets(
-    TARGETS cuspatial
-    DESTINATION ${cython_lib_dir}
-  )
+  install(TARGETS cuspatial DESTINATION ${cython_lib_dir})
 endif()
 
 include(rapids-cython-core)

diff --git a/python/cuspatial/cmake/Modules/WheelHelpers.cmake b/python/cuspatial/cmake/Modules/WheelHelpers.cmake
diff --git a/python/cuspatial/cuspatial/__init__.py b/python/cuspatial/cuspatial/__init__.py
@@ -1,4 +1,8 @@
-# Copyright (c) 2023, NVIDIA CORPORATION.
+# Copyright (c) 2023-2024, NVIDIA CORPORATION.
+
+# load cudf before any other imports, to be sure libcudf's symbols are found
+# in the libcudf.so from the libcudf wheel (if one is installed)
+import cudf
 
 from ._version import __git_commit__, __version__
 from .core.geodataframe import GeoDataFrame