fix

rapidsai · Mar 21, 2024 · 8a01dd9 · 8a01dd9
1 parent b595308
commit 8a01dd9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/cuvs/cuvs/neighbors/brute_force/__init__.py b/python/cuvs/cuvs/neighbors/brute_force/__init__.py
@@ -13,6 +13,6 @@
 # limitations under the License.
 
 
-from .brute_force import Index, build, search
+from .brute_force import FlatIndex, build, search
 
-__all__ = ["Index", "build", "search"]
+__all__ = ["FlatIndex", "build", "search"]
diff --git a/python/cuvs/cuvs/neighbors/brute_force/brute_force.pyx b/python/cuvs/cuvs/neighbors/brute_force/brute_force.pyx
@@ -19,7 +19,7 @@ import numpy as np
 
 cimport cuvs.common.cydlpack
 
-from cuvs.common.Resources import auto_sync_resources
+from cuvs.common.resources import auto_sync_resources
 
 from cython.operator cimport dereference as deref
 from libc.stdint cimport uint32_t