diff options
author | Arne Juul <arnej@yahooinc.com> | 2023-04-26 13:56:05 +0000 |
---|---|---|
committer | Arne Juul <arnej@yahooinc.com> | 2023-04-27 13:27:36 +0000 |
commit | 7d1da87a9165e7d9aca41c546ea7b0a370a960b8 (patch) | |
tree | 839d00add49c061d7c64860bc19a571680f1dcfe /searchlib/src/tests/queryeval | |
parent | f5dd3cb5d31875cf596adc01f2207f690afe553f (diff) |
remove unused distance functions
Diffstat (limited to 'searchlib/src/tests/queryeval')
-rw-r--r-- | searchlib/src/tests/queryeval/nearest_neighbor/nearest_neighbor_test.cpp | 18 |
1 files changed, 4 insertions, 14 deletions
diff --git a/searchlib/src/tests/queryeval/nearest_neighbor/nearest_neighbor_test.cpp b/searchlib/src/tests/queryeval/nearest_neighbor/nearest_neighbor_test.cpp index fd07529795a..b9599a0c75d 100644 --- a/searchlib/src/tests/queryeval/nearest_neighbor/nearest_neighbor_test.cpp +++ b/searchlib/src/tests/queryeval/nearest_neighbor/nearest_neighbor_test.cpp @@ -29,7 +29,6 @@ using search::attribute::DistanceMetric; using search::feature_t; using search::tensor::DenseTensorAttribute; using search::tensor::DistanceCalculator; -using search::tensor::DistanceFunction; using search::tensor::SerializedFastValueAttribute; using search::tensor::TensorAttribute; using vespalib::eval::CellType; @@ -49,9 +48,6 @@ vespalib::string denseSpecDouble("tensor(x[2])"); vespalib::string denseSpecFloat("tensor<float>(x[2])"); vespalib::string mixed_spec("tensor(m{},x[2])"); -DistanceFunction::UP euclid_d = search::tensor::make_distance_function(DistanceMetric::Euclidean, CellType::DOUBLE); -DistanceFunction::UP euclid_f = search::tensor::make_distance_function(DistanceMetric::Euclidean, CellType::FLOAT); - std::unique_ptr<Value> createTensor(const TensorSpec &spec) { return SimpleValue::from_spec(spec); } @@ -119,14 +115,6 @@ struct Fixture { auto t = createTensor(_typeSpec, v1, v2); setTensor(docId, *t); } - - const DistanceFunction &dist_fun() const { - if (_cfg.tensorType().cell_type() == CellType::FLOAT) { - return *euclid_f; - } else { - return *euclid_d; - } - } }; template <bool strict> @@ -136,9 +124,11 @@ SimpleResult find_matches(Fixture &env, const Value &qtv, double threshold = std auto &attr = *(env._attr); auto dff = search::tensor::make_distance_function_factory(DistanceMetric::Euclidean, qtv.cells().type); - DistanceCalculator dist_calc(attr, dff->for_query_vector(qtv.cells())); + auto df = dff->for_query_vector(qtv.cells()); + threshold = df->convert_threshold(threshold); + DistanceCalculator dist_calc(attr, std::move(df)); NearestNeighborDistanceHeap dh(2); - dh.set_distance_threshold(env.dist_fun().convert_threshold(threshold)); + dh.set_distance_threshold(threshold); const GlobalFilter &filter = *env._global_filter; auto search = NearestNeighborIterator::create(strict, tfmd, dist_calc, dh, filter); if (strict) { |