Re-enable cloud test

Author: Nikos Papailiou

apis/python/src/tiledb/vector_search/ingestion.py

@@ -937,10 +937,7 @@ def centralised_kmeans(
     ):
         from sklearn.cluster import KMeans
 
-        from tiledb.vector_search.module import (
-            array_to_matrix,
-            kmeans_fit,
-        )
+        from tiledb.vector_search.module import array_to_matrix
 
         with tiledb.scope_ctx(ctx_or_config=config):
             logger = setup(config, verbose)
@@ -990,6 +987,7 @@ def centralised_kmeans(
                     km.fit_predict(sample_vectors)
                     centroids = np.transpose(np.array(km.cluster_centers_))
                 else:
+                    from tiledb.vector_search.module import kmeans_fit
                     centroids = kmeans_fit(partitions, init, max_iter, verbose, n_init, array_to_matrix(np.transpose(sample_vectors)))
                     centroids = np.array(centroids) # TODO: why is this here?
             else:

apis/python/test/test_cloud.py

@@ -83,10 +83,7 @@ def test_cloud_ivf_flat(self):
             partitions=partitions,
             input_vectors_per_work_item=5000,
             config=tiledb.cloud.Config().dict(),
-            # TODO Re-enable.
-            #  This is temporarily disabled due to an incompatibility of new ingestion code and previous
-            #  UDF library releases.
-            # mode=Mode.BATCH,
+            mode=Mode.BATCH,
         )
 
         tiledb_index_uri = groups.info(index_uri).tiledb_uri