From cf88458e4bd6e1df77f6c605fc9e09f7aa61d438 Mon Sep 17 00:00:00 2001 From: Kaiyu Yang Date: Wed, 12 Oct 2022 22:33:48 -0400 Subject: [PATCH] Fix a typo in arrow_dataset.py --- src/datasets/arrow_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/datasets/arrow_dataset.py b/src/datasets/arrow_dataset.py index a0b8481825d..d3db912357b 100644 --- a/src/datasets/arrow_dataset.py +++ b/src/datasets/arrow_dataset.py @@ -4705,7 +4705,7 @@ def add_faiss_index( This is passed to the index factory of Faiss to create the index. Default index class is ``IndexFlat``. metric_type (Optional :obj:`int`): - Type of metric. Ex: faiss.faiss.METRIC_INNER_PRODUCT or faiss.METRIC_L2. + Type of metric. Ex: faiss.METRIC_INNER_PRODUCT or faiss.METRIC_L2. custom_index (Optional :obj:`faiss.Index`): Custom Faiss index that you already have instantiated and configured for your needs. batch_size (Optional :obj:`int`): Size of the batch to use while adding vectors to the FaissIndex. Default value is 1000.