maybe fix milvus & flann, disable kgraph

erikbern · Apr 7, 2023 · 28cdde8 · 28cdde8
1 parent 72f50e2
commit 28cdde8
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/.github/workflows/benchmarks.yml b/.github/workflows/benchmarks.yml
@@ -28,7 +28,6 @@ jobs:
           - faiss
           - flann
           - hnswlib
-          - kgraph
           - luceneknn
           - milvus
           - mrpt

diff --git a/algos.yaml b/algos.yaml
@@ -630,6 +630,7 @@ float:
           args: [[100], [.nan], [4], ["squared_l2"]]
           query-args: [[[2, 20], [3, 20], [3, 30], [4, 30], [5, 40], [8, 80]]]
     kgraph:
+      disable: true
       docker-tag: ann-benchmarks-kgraph
       module: ann_benchmarks.algorithms.kgraph
       constructor: KGraph
@@ -929,6 +930,7 @@ float:
             ]
             query-args: [[0.1, 0.2, 0.5, 0.7, 0.9, 0.95, 0.99]]
     kgraph:
+      disable: true
       docker-tag: ann-benchmarks-kgraph
       module: ann_benchmarks.algorithms.kgraph
       constructor: KGraph
@@ -1163,6 +1165,7 @@ bit:
           args: { "d" : "@dimension", "r" : [0, 2, 3],
             "chunk-factor" : [0.2, 0.5, 0.8] }
     kgraph:
+      disable: true
       docker-tag: ann-benchmarks-kgraph
       module: ann_benchmarks.algorithms.kgraph
       constructor: KGraph

diff --git a/install/Dockerfile.flann b/install/Dockerfile.flann
@@ -6,5 +6,4 @@ RUN mkdir flann/build
 RUN cd flann/build && cmake ..
 RUN cd flann/build && make -j4
 RUN cd flann/build && make install
-RUN pip3 install sklearn
 RUN python3 -c 'import pyflann'
diff --git a/install/Dockerfile.milvus b/install/Dockerfile.milvus
@@ -5,7 +5,7 @@ COPY run_algorithm.py requirements.txt ./
 
 # update python packages
 RUN pip3 install -r requirements.txt
+RUN pip3 install scipy==1.10.1
 RUN python3 -c 'import pyknowhere'
 
 ENTRYPOINT ["python3", "-u", "run_algorithm.py"]
-
-Original file line number
+Diff line change
@@ Expand Up / @@ -28,7 +28,6 @@ jobs: @@
               - faiss
               - flann
               - hnswlib
-              - kgraph
               - luceneknn
               - milvus
               - mrpt
@@ Expand Down @@