add #[pymethods]

rigetti · Feb 29, 2024 · f752f64 · f752f64
1 parent 4821de4
commit f752f64
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/quil-py/src/program/mod.rs b/quil-py/src/program/mod.rs
@@ -35,7 +35,9 @@ use crate::{
 
 use self::{
     analysis::{PyBasicBlock, PyControlFlowGraph},
-    scheduling::{PyScheduleSeconds, PyScheduleSecondsItem, PyTimeSpanSeconds},
+    scheduling::{
+        PyScheduleSeconds, PyScheduleSecondsItem, PyScheduledBasicBlock, PyTimeSpanSeconds,
+    },
 };
 pub use self::{calibration::PyCalibrationSet, frame::PyFrameSet, memory::PyMemoryRegion};
 
@@ -373,5 +375,5 @@ impl PyProgram {
 }
 
 create_init_submodule! {
-    classes: [ PyFrameSet, PyProgram, PyCalibrationSet, PyMemoryRegion, PyBasicBlock, PyControlFlowGraph, PyScheduleSeconds, PyScheduleSecondsItem, PyTimeSpanSeconds ],
+    classes: [ PyFrameSet, PyProgram, PyCalibrationSet, PyMemoryRegion, PyBasicBlock, PyControlFlowGraph, PyScheduledBasicBlock, PyScheduleSeconds, PyScheduleSecondsItem, PyTimeSpanSeconds ],
 }
diff --git a/quil-py/src/program/scheduling.rs b/quil-py/src/program/scheduling.rs
@@ -23,6 +23,7 @@ py_wrap_error!(
     PyValueError
 );
 
+#[pymethods]
 impl PyScheduledBasicBlock {
     pub fn as_schedule_seconds(&self, program: &PyProgram) -> PyResult<PyScheduleSeconds> {
         ScheduledBasicBlock::from(&self.0)