Skip to content

Commit d0aa442

Browse files
committed
fix benchmark.yaml and docker-tag within configs to point to folder specific dockerfiles
1 parent d56a0dc commit d0aa442

File tree

8 files changed

+19
-13
lines changed

8 files changed

+19
-13
lines changed

.github/workflows/benchmarks.yml

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,28 +30,34 @@ jobs:
3030
dataset: [random-xs-20-angular]
3131
library:
3232
- annoy
33+
- balltree
34+
- bruteforce
35+
- ckdtree
3336
- diskann
34-
- dolphinn
37+
- dolphinnpy
3538
- elasticsearch
3639
- elastiknn
3740
- faiss
3841
- flann
3942
- glass
4043
- hnswlib
44+
- kdtree
4145
- luceneknn
4246
- milvus
4347
- mrpt
4448
- n2
4549
- ngt
4650
- nmslib
51+
- onng_ngt
4752
- opensearchknn
53+
- panng_ngt
4854
- pgvector
4955
- pynndescent
5056
- redisearch
5157
- qdrant
52-
- qsgngt
58+
- qg_ngt
59+
- qsg_ngt
5360
- scann
54-
- scipy
5561
- sklearn
5662
- sptag
5763
- tinyknn

ann_benchmarks/algorithms/balltree/config.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ float:
33
- base_args: ['@metric']
44
constructor: BallTree
55
disabled: true
6-
docker_tag: ann-benchmarks-sklearn
6+
docker_tag: ann-benchmarks-balltree
77
module: ann_benchmarks.algorithms.balltree
88
name: ball
99
run_groups:

ann_benchmarks/algorithms/bruteforce/config.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ bit:
33
- base_args: ['@metric']
44
constructor: BruteForceBLAS
55
disabled: false
6-
docker_tag: ann-benchmarks-sklearn
6+
docker_tag: ann-benchmarks-bruteforce
77
module: ann_benchmarks.algorithms.bruteforce
88
name: bf
99
run_groups:
@@ -14,7 +14,7 @@ float:
1414
- base_args: ['@metric']
1515
constructor: BruteForce
1616
disabled: true
17-
docker_tag: ann-benchmarks-sklearn
17+
docker_tag: ann-benchmarks-bruteforce
1818
module: ann_benchmarks.algorithms.bruteforce
1919
name: bruteforce
2020
run_groups:
@@ -23,7 +23,7 @@ float:
2323
- base_args: ['@metric']
2424
constructor: BruteForceBLAS
2525
disabled: false
26-
docker_tag: ann-benchmarks-sklearn
26+
docker_tag: ann-benchmarks-bruteforce
2727
module: ann_benchmarks.algorithms.bruteforce
2828
name: bruteforce-blas
2929
run_groups:

ann_benchmarks/algorithms/ckdtree/config.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ float:
33
- base_args: ['@metric']
44
constructor: CKDTree
55
disabled: true
6-
docker_tag: ann-benchmarks-scipy
6+
docker_tag: ann-benchmarks-ckdtree
77
module: ann_benchmarks.algorithms.ckdtree
88
name: ckdtree
99
run_groups:

ann_benchmarks/algorithms/dolphinnpy/config.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ float:
33
- base_args: {}
44
constructor: DolphinnPy
55
disabled: true
6-
docker_tag: ann-benchmarks-dolphinn
6+
docker_tag: ann-benchmarks-dolphinnpy
77
module: ann_benchmarks.algorithms.dolphinnpy
88
name: DolphinnPy
99
run_groups:

ann_benchmarks/algorithms/dummy_algo/config.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ float:
33
- base_args: ['@metric']
44
constructor: DummyAlgoSt
55
disabled: true
6-
docker_tag: ann-benchmarks-sklearn
6+
docker_tag: ann-benchmarks-dummy_algo
77
module: ann_benchmarks.algorithms.dummy_algo
88
name: dummy-algo-st
99
run_groups:
@@ -12,7 +12,7 @@ float:
1212
- base_args: ['@metric']
1313
constructor: DummyAlgoMt
1414
disabled: true
15-
docker_tag: ann-benchmarks-sklearn
15+
docker_tag: ann-benchmarks-dummy_algo
1616
module: ann_benchmarks.algorithms.dummy_algo
1717
name: dummy-algo-mt
1818
run_groups:

ann_benchmarks/algorithms/kdtree/config.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ float:
33
- base_args: ['@metric']
44
constructor: KDTree
55
disabled: false
6-
docker_tag: ann-benchmarks-sklearn
6+
docker_tag: ann-benchmarks-kdtree
77
module: ann_benchmarks.algorithms.kdtree
88
name: kd
99
run_groups:

ann_benchmarks/algorithms/qsg_ngt/config.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ float:
33
- base_args: ['@metric', Float, 0.1]
44
constructor: QSG
55
disabled: false
6-
docker_tag: ann-benchmarks-qsgngt
6+
docker_tag: ann-benchmarks-qsg_ngt
77
module: ann_benchmarks.algorithms.qsg_ngt
88
name: qsgngt
99
run_groups:

0 commit comments

Comments
 (0)