From 8a01dd96906e64e79da6044f765a22d05de05d01 Mon Sep 17 00:00:00 2001 From: Ben Frederickson Date: Thu, 21 Mar 2024 14:24:25 -0700 Subject: [PATCH] fix --- python/cuvs/cuvs/neighbors/brute_force/__init__.py | 4 ++-- python/cuvs/cuvs/neighbors/brute_force/brute_force.pyx | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/python/cuvs/cuvs/neighbors/brute_force/__init__.py b/python/cuvs/cuvs/neighbors/brute_force/__init__.py index b88c4b464..fadc186a2 100644 --- 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 index 1bc4fc1c1..3256ab923 100644 --- 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