Also adding definition for UnitaryGate

alexanderivrii · Feb 10, 2025 · f72152e · f72152e
1 parent 1108536
commit f72152e
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/crates/circuit/src/operations.rs b/crates/circuit/src/operations.rs
@@ -2848,7 +2848,19 @@ impl Operation for UnitaryGate {
         }
     }
     fn definition(&self, _params: &[Param]) -> Option<CircuitData> {
-        None
+        Python::with_gil(|py| -> Option<CircuitData> {
+            let py_op = self
+                .create_py_op(py, &ExtraInstructionAttributes::default())
+                .unwrap();
+            match py_op.getattr(py, intern!(py, "definition")) {
+                Ok(definition) => definition
+                    .getattr(py, intern!(py, "_data"))
+                    .ok()?
+                    .extract::<CircuitData>(py)
+                    .ok(),
+                Err(_) => None,
+            }
+        })
     }
     fn standard_gate(&self) -> Option<StandardGate> {
         None