From f470d86260426d0a7c0fba776f322718579eb2d6 Mon Sep 17 00:00:00 2001
From: divyegala <divyegala@gmail.com>
Date: Mon, 30 Oct 2023 11:14:30 -0700
Subject: [PATCH] add wiki datasets to datasets.yaml

---
 .../src/raft-ann-bench/run/conf/datasets.yaml  | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/python/raft-ann-bench/src/raft-ann-bench/run/conf/datasets.yaml b/python/raft-ann-bench/src/raft-ann-bench/run/conf/datasets.yaml
index 45d648f88e..33f07b9f22 100644
--- a/python/raft-ann-bench/src/raft-ann-bench/run/conf/datasets.yaml
+++ b/python/raft-ann-bench/src/raft-ann-bench/run/conf/datasets.yaml
@@ -83,3 +83,21 @@
   query_file: sift-128-euclidean/query.fbin
   groundtruth_neighbors_file: sift-128-euclidean/groundtruth.neighbors.ibin
   distance: euclidean
+
+- name: wiki_all_1M,
+  base_file: wiki_all_1M/base.88M.fbin,
+  query_file: wiki_all_1M/queries.fbin,
+  groundtruth_neighbors_file: wiki_all_1M/groundtruth.88M.neighbors.ibin,
+  distance: euclidean
+
+- name: wiki_all_10M,
+  base_file: wiki_all_10M/base.88M.fbin,
+  query_file: wiki_all_10M/queries.fbin,
+  groundtruth_neighbors_file: wiki_all_10M/groundtruth.88M.neighbors.ibin,
+  distance: euclidean
+
+- name: wiki_all_88M,
+  base_file: wiki_all_88M/base.88M.fbin,
+  query_file: wiki_all_88M/queries.fbin,
+  groundtruth_neighbors_file: wiki_all_88M/groundtruth.88M.neighbors.ibin,
+  distance: euclidean