Updating to pyo3 0.22

PECOS-packages · Oct 7, 2024 · 8415a8c · 8415a8c
1 parent fcbded1
commit 8415a8c
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 33 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -21,7 +21,7 @@ keywords = ["scientific", "quantum", "QEC"]
 categories = ["science", "simulation"]
 
 [workspace.dependencies]
-pyo3 = "0.20"  # TODO: Upgrade to 0.22!
+pyo3 = "0.22"
 rand = "0.8"
 rand_chacha = "0.3"
 rand_xoshiro = "0.6"

diff --git a/crates/pecos-python/src/lib.rs b/crates/pecos-python/src/lib.rs
@@ -204,11 +204,12 @@ impl SparseSim {
     }
 
     #[allow(clippy::too_many_lines)]
+    #[pyo3(signature = (symbol, location, params=None))]
     fn run_gate(
         &mut self,
         symbol: &str,
-        location: &PyTuple,
-        params: Option<&PyDict>,
+        location: &Bound<'_, PyTuple>,
+        params: Option<&Bound<'_, PyDict>>,
     ) -> PyResult<Option<HashMap<usize, u8>>> {
         match (symbol, location.len()) {
             ("X", 1) => {
@@ -453,6 +454,7 @@ impl SparseSim {
         self.inner.destab_tableau()
     }
 
+    #[pyo3(signature = (verbose=None, _print_y=None, print_destabs=None))]
     fn print_stabs(
         &self,
         verbose: Option<bool>,
@@ -496,7 +498,7 @@ impl SparseSim {
 }
 
 #[pymodule]
-fn _pecos_rslib(_py: Python, m: &PyModule) -> PyResult<()> {
+fn _pecos_rslib(_py: Python<'_>, m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_class::<SparseSim>()?;
     Ok(())
 }