feat(python): define default arguments to methods

rigetti · Jan 10, 2023 · 1ae1298 · 1ae1298
1 parent e847da7
commit 1ae1298
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/crates/python/src/executable.rs b/crates/python/src/executable.rs
@@ -43,6 +43,14 @@ pub struct PyParameter {
 #[pymethods]
 impl PyExecutable {
     #[new]
+    #[args(
+        "/",
+        registers = "Vec::new()",
+        parameters = "Vec::new()",
+        shots = "None",
+        compile_with_quilc = "None",
+        compiler_options = "None"
+    )]
     pub fn new(
         quil: String,
         registers: Vec<String>,

diff --git a/crates/python/src/qpu/client.rs b/crates/python/src/qpu/client.rs
@@ -48,6 +48,7 @@ py_wrap_type! {
 impl PyQcsClient {
     // TODO: default arg
     #[new]
+    #[args("/", use_gateway = "None")]
     pub fn new(py: Python<'_>, use_gateway: Option<bool>) -> PyResult<Self> {
         future_into_py(py, async move {
             let client = Qcs::load()

diff --git a/crates/python/src/qpu/quilc/mod.rs b/crates/python/src/qpu/quilc/mod.rs
@@ -21,6 +21,7 @@ py_wrap_type! {
 #[pymethods]
 impl PyCompilerOpts {
     #[new]
+    #[args("/", timeout = "DEFAULT_COMPILER_TIMEOUT")]
     pub fn new(timeout: Option<u8>) -> Self {
         Self::from(CompilerOpts::new().with_timeout(timeout))
     }