Tiny refactor of query graph remove_nodes method

This commit is contained in:
Loïc Lecrenier
2023-03-30 14:49:25 +02:00
parent 0d6e8b5c31
commit 061b1e6d7c
3 changed files with 48 additions and 21 deletions

View File

@ -76,7 +76,7 @@ impl<'ctx> RankingRule<'ctx, QueryGraph> for Words {
self.exhausted = true;
} else {
let nodes_to_remove = self.nodes_to_remove.pop().unwrap();
query_graph.remove_nodes(&nodes_to_remove.iter().collect::<Vec<_>>());
query_graph.remove_nodes_keep_edges(&nodes_to_remove.iter().collect::<Vec<_>>());
}
Ok(Some(RankingRuleOutput { query: child_query_graph, candidates: this_bucket }))