Revert dataset.py changes

westonpace · Jun 20, 2024 · 062ac38 · 062ac38
1 parent cb34966
commit 062ac38
Showing 1 changed file with 0 additions and 9 deletions.
diff --git a/python/python/lance/dataset.py b/python/python/lance/dataset.py
@@ -40,7 +40,6 @@
 )
 from .dependencies import numpy as np
 from .dependencies import pandas as pd
-from .file import LanceFileWriter
 from .fragment import FragmentMetadata, LanceFragment
 from .lance import (
     CleanupStats,
@@ -1487,14 +1486,6 @@ def create_index(
                     metric,
                     accelerator,
                 )
-                arrow_ivf_centroids = pa.FixedSizeListArray.from_arrays(
-                    ivf_centroids.reshape(-1), ivf_centroids.shape[1]
-                )
-                tab = pa.table({"ivf_centroids": arrow_ivf_centroids})
-                with LanceFileWriter(
-                    "/tmp/ivf_centroids.lance", schema=tab.schema
-                ) as f:
-                    f.write_batch(tab)
                 kwargs["precomputed_partitions_file"] = partitions_file
 
             if (ivf_centroids is None) and (pq_codebook is not None):