Update umap_utils.pyx

fix import
rapidsai · Oct 7, 2024 · aec5743 · aec5743
1 parent df3f87a
commit aec5743
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cuml/cuml/manifold/umap_utils.pyx b/python/cuml/cuml/manifold/umap_utils.pyx
@@ -16,7 +16,7 @@
 
 # distutils: language = c++
 
-from rmm.librmm.memory_resource cimport get_current_device_resource
+from rmm.pylibrmm.memory_resource cimport get_current_device_resource
 from pylibraft.common.handle cimport handle_t
 from cuml.manifold.umap_utils cimport *
 from cuml.metrics.distance_type cimport DistanceType