Merge branch 'move-basis-search-dijkstra' into move-basis-search

raynelfss · Jul 21, 2024 · e746de1 · e746de1
2 parents 3096871 + 983ceaf
commit e746de1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/crates/circuit/src/equivalence.rs b/crates/circuit/src/equivalence.rs
@@ -323,7 +323,7 @@ impl ToPyObject for CircuitRep {
 }
 
 // Custom Types
-type GraphType = StableDiGraph<NodeData, EdgeData>;
+type GraphType = StableDiGraph<NodeData, Option<EdgeData>>;
 type KTIType = HashMap<Key, NodeIndex>;
 
 #[pyclass(
@@ -619,7 +619,7 @@ impl EquivalenceLibrary {
             )
         }));
         for edge in edges {
-            self.graph.add_edge(edge.0, edge.1, edge.2);
+            self.graph.add_edge(edge.0, edge.1, Some(edge.2));
         }
         self.rule_id += 1;
         self._graph = None;