Skip to content

Commit 12517dc

Browse files
[8.0] Flow test improvements (#587)
Flow test improvements (#585) * fix hnsw parallel tests * update pyproject toml file (cherry picked from commit bcb3033) Co-authored-by: GuyAv46 <[email protected]>
1 parent 8f10c26 commit 12517dc

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

pyproject.toml

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,9 @@ classifiers = [
1818
'Programming Language :: Python :: 3.9',
1919
'Programming Language :: Python :: 3.10',
2020
'Programming Language :: Python :: 3.11',
21+
'Programming Language :: Python :: 3.12',
22+
'Programming Language :: Python :: 3.13',
23+
'Programming Language :: Python :: 3.14',
2124
'License :: OSI Approved :: BSD License',
2225
'Development Status :: 3 - Alpha',
2326
]
@@ -28,8 +31,8 @@ python = "^3.10"
2831
[tool.poetry.dev-dependencies]
2932
numpy = "*"
3033
hnswlib = "^0.6.2"
31-
pytest = "^6.2.4"
32-
scipy = "^1.10.1"
34+
pytest = "^8.3"
35+
scipy = "^1.14.1"
3336
h5py = "^3.7.0"
3437
ml_dtypes = "*"
3538

tests/flow/test_hnsw_parallel.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
# Helper class for creating a "baseline" HNSW index that was built by inserting vectors one by one and a corresponding
1212
# flat index for given params, to compare the parallel operation against it
13-
class TestIndex:
13+
class IndexTest:
1414
def __init__(self, dim_, num_elements_, metric_, data_type_, multi_=False, ef_runtime=200):
1515

1616
self.dim = dim_
@@ -81,10 +81,10 @@ def compute_ground_truth_range(self, query_data, radius):
8181
per_label = 5 # for multi value index
8282

8383
print("Creating test indexes...")
84-
g_test_index = TestIndex(dim, num_elements, metric, data_type)
84+
g_test_index = IndexTest(dim, num_elements, metric, data_type)
8585
g_test_index.insert_random_vectors()
8686

87-
g_test_index_multi = TestIndex(dim, num_elements, metric, data_type, multi_=True)
87+
g_test_index_multi = IndexTest(dim, num_elements, metric, data_type, multi_=True)
8888
g_test_index_multi.insert_random_vectors_multi(per_label)
8989

9090
def test_parallel_search():

0 commit comments

Comments
 (0)