Merge pull request #6 from shuninoz/master

add partitions col support
jelmerk · Jan 14, 2025 · b73d282 · b73d282
2 parents 662691a + c4e923f
commit b73d282
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/hnswlib-spark/src/main/python/pyspark_hnsw/knn.py b/hnswlib-spark/src/main/python/pyspark_hnsw/knn.py
@@ -511,6 +511,7 @@ class HnswSimilarity(JavaEstimator, _HnswParams, JavaMLReadable, JavaMLWritable)
     def __init__(
         self,
         identifierCol="id",
+        partitionCol=None,
         queryIdentifierCol=None,
         queryPartitionsCol=None,
         parallelism=None,