Mirrors/the-algorithm
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm.git synced 2024-06-13 14:48:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge 25b23e0f86 into 138bb51997

Browse Source
This commit is contained in:
Anzori (Nika) Ghurtchumelia 2023-04-04 21:09:49 -05:00 committed by GitHub
commit f1b3792445
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ann/src/main/scala/com/twitter/ann/hnsw/JMapBasedIdEmbeddingMap.scala
View File

@ -44,10 +44,7 @@ private[hnsw] object JMapBasedIdEmbeddingMap {
injection: Injection[T, Array[Byte]],
numElements: Option[Int] = Option.empty
): IdEmbeddingMap[T] = {
val map = numElements match {
case Some(elements) => new ConcurrentHashMap[T, EmbeddingVector](elements)
case None => new ConcurrentHashMap[T, EmbeddingVector]()
}
val map = numElements.fold(new ConcurrentHashMap[T, EmbeddingVector])(elems => new ConcurrentHashMap(elems))
HnswIOUtil.loadEmbeddings(
embeddingFile,
injection,