Fix vector search import (#1287)

chanzuckerberg · Sep 26, 2024 · af68b5e · af68b5e
1 parent dc4eb1f
commit af68b5e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/api/python/cellxgene_census/src/cellxgene_census/experimental/_embedding_search.py b/api/python/cellxgene_census/src/cellxgene_census/experimental/_embedding_search.py
@@ -8,7 +8,6 @@
 import numpy as np
 import numpy.typing as npt
 import pandas as pd
-import tiledb.vector_search as vs
 import tiledbsoma as soma
 from scipy import sparse
 
@@ -70,6 +69,8 @@ def find_nearest_obs(
             The result of `get_embedding_metadata_by_name(embedding_name, organism, census_version)`.
             Supplying this saves a network request for repeated searches.
     """
+    import tiledb.vector_search as vs
+
     if embedding_metadata is None:
         embedding_metadata = get_embedding_metadata_by_name(embedding_name, organism, census_version)
     assert embedding_metadata["embedding_name"] == embedding_name