diff --git a/Sources/SwiftNLP/1. Data Collection/HNSWCorpus.swift b/Sources/SwiftNLP/1. Data Collection/HNSWCorpus.swift index cd7c7272bbd53f1bac2f3333b8001ff0eaa7b217..da0cf18382db380fc193499c5473e712630cf518 100644 --- a/Sources/SwiftNLP/1. Data Collection/HNSWCorpus.swift +++ b/Sources/SwiftNLP/1. Data Collection/HNSWCorpus.swift @@ -27,7 +27,7 @@ class HNSWCorpus<Scalar: BinaryFloatingPoint & Codable>: SNLPCorpus { // internal var _documentEncoder: ContextFreeEncoder<Scalar> internal var _documentEncoder: any SNLPEncoder - var zeroes: [Scalar] + var zeroes: [Scalar] { _documentEncoder.zeroes as! [Scalar] } var encodedDocuments: DeterministicSampleVectorIndex<[Scalar]> var count: Int { encodedDocuments.base.vectors.count } @@ -36,13 +36,11 @@ class HNSWCorpus<Scalar: BinaryFloatingPoint & Codable>: SNLPCorpus { init(encoding: ContextFreeEncoder<Scalar>.PreComputedEmbeddings, scalar: Scalar.Type = Double.self, typicalNeighborhoodSize: Int = 20) { _documentEncoder = ContextFreeEncoder(source: encoding) - zeroes = _documentEncoder.zeroes as! [Scalar] encodedDocuments = DeterministicSampleVectorIndex<[Scalar]>(typicalNeighborhoodSize: typicalNeighborhoodSize) } init(encoder: any SNLPEncoder, scalar: Scalar.Type = Double.self, typicalNeighborhoodSize: Int = 20) { _documentEncoder = encoder - zeroes = _documentEncoder.zeroes as! [Scalar] encodedDocuments = DeterministicSampleVectorIndex<[Scalar]>(typicalNeighborhoodSize: typicalNeighborhoodSize) }