Stub pyi file autogen using pyo3-stub-gen

python/Cargo.toml

@@ -11,7 +11,7 @@ readme = "README.md"
 
 [lib]
 name = "egobox"
-crate-type = ["cdylib"]
+crate-type = ["cdylib", "rlib"]
 
 [features]
 default = []
@@ -48,3 +48,8 @@ serde_json.workspace = true
 ctrlc.workspace = true
 
 argmin_testfunctions.workspace = true
+pyo3-stub-gen = { path = "../../pyo3-stub-gen/pyo3-stub-gen", features = ["numpy"] }
+
+[[bin]]
+name = "stub_gen"
+doc = false

python/pyproject.toml

@@ -21,7 +21,8 @@ build-backend = "maturin"
 requires = ["maturin>=1.0, <2.0"]
 
 [tool.maturin]
-python-source = "./"
+python-source = "egobox"
+module-name = "egobox"
 features = ["pyo3/extension-module"]
 # Optional usage of BLAS backend 
 # cargo-extra-args = "--features linfa/intel-mkl-static"

python/src/bin/stub_gen.rs

@@ -0,0 +1,8 @@
+use pyo3_stub_gen::Result;
+
+fn main() -> Result<()> {
+    env_logger::Builder::from_env(env_logger::Env::default().filter_or("RUST_LOG", "info")).init();
+    let stub = egobox::stub_info()?;
+    stub.generate()?;
+    Ok(())
+}

python/src/egor.rs

@@ -16,6 +16,7 @@ use ndarray::{concatenate, Array1, Array2, ArrayView2, Axis};
 use numpy::{IntoPyArray, PyArray1, PyArray2, PyArrayMethods, PyReadonlyArray2, ToPyArray};
 use pyo3::exceptions::PyValueError;
 use pyo3::prelude::*;
+use pyo3_stub_gen::derive::{gen_stub_pyclass, gen_stub_pyfunction, gen_stub_pymethods};
 
 /// Utility function converting `xlimits` float data list specifying bounds of x components
 /// to x specified as a list of XType.Float types [egobox.XType]
@@ -25,6 +26,7 @@ use pyo3::prelude::*;
 ///
 /// # Returns
 ///     xtypes: nx-size list of XSpec(XType(FLOAT), [lower_bound, upper_bounds]) where `nx` is the dimension of x
+#[gen_stub_pyfunction]
 #[pyfunction]
 pub(crate) fn to_specs(py: Python, xlimits: Vec<Vec<f64>>) -> PyResult<PyObject> {
     if xlimits.is_empty() || xlimits[0].is_empty() {
@@ -149,6 +151,7 @@ pub(crate) fn to_specs(py: Python, xlimits: Vec<Vec<f64>>) -> PyResult<PyObject>
 ///     seed (int >= 0)
 ///         Random generator seed to allow computation reproducibility.
 ///      
+#[gen_stub_pyclass]
 #[pyclass]
 pub(crate) struct Egor {
     pub xspecs: PyObject,
@@ -174,6 +177,7 @@ pub(crate) struct Egor {
     pub seed: Option<u64>,
 }
 
+#[gen_stub_pyclass]
 #[pyclass]
 pub(crate) struct OptimResult {
     #[pyo3(get)]
@@ -186,6 +190,7 @@ pub(crate) struct OptimResult {
     y_doe: Py<PyArray2<f64>>,
 }
 
+#[gen_stub_pymethods]
 #[pymethods]
 impl Egor {
     #[new]