Merge pull request #32 from TrevorHansen/om14

Fix faster-greedy-dag extractor.
egraphs-good · Jan 1, 2024 · 80d9c63 · 80d9c63
2 parents 4c377b7 + b9c1228
commit 80d9c63
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/extract/faster_greedy_dag.rs b/src/extract/faster_greedy_dag.rs
@@ -22,11 +22,11 @@ impl FasterGreedyDagExtractor {
         best_cost: Cost,
     ) -> CostSet {
         let node = &egraph[&node_id];
+        let cid = egraph.nid_to_cid(&node_id);
 
-        // No children -> easy.
         if node.children.is_empty() {
             return CostSet {
-                costs: Default::default(),
+                costs: HashMap::from([(cid.clone(), node.cost)]),
                 total: node.cost,
                 choice: node_id.clone(),
             };
@@ -41,8 +41,6 @@ impl FasterGreedyDagExtractor {
         childrens_classes.sort();
         childrens_classes.dedup();
 
-        let cid = egraph.nid_to_cid(&node_id);
-
         let first_cost = costs.get(&childrens_classes[0]).unwrap();
 
         if childrens_classes.contains(cid)