add _ prefix to extension module

Author: sevas

cg_rustpy/proj_rs.py

@@ -0,0 +1,6 @@
+from numpy.typing import NDArray
+from ._cg_rustpy import project_points_rs as _project_points_rs
+
+
+def project_points_rs(points: NDArray, k: NDArray, dist_coeffs: NDArray):
+    return _project_points_rs(points, k, dist_coeffs)

pyproject.toml

@@ -22,6 +22,7 @@ dependencies = [
  ]
 
 
+
 [project.optional-dependencies]
 mlx = ["mlx"]
 dev = ["ruff", "pytest", "pytest-benchmark", "jupyterlab"]

rust/src/lib.rs

@@ -1,4 +1,4 @@
-#[pyo3::pymodule]
+#[pyo3::pymodule(name = "_cg_rustpy")]
 mod cg_rustpy {
     use numpy::ndarray::{
         Array, Array2, ArrayView1, ArrayView2

setup.py

@@ -7,7 +7,7 @@
     packages=["cg"],
     rust_extensions=[
         RustExtension(
-            "cg_rustpy.cg_rustpy", binding=Binding.PyO3, path="rust/Cargo.toml"
+            "cg_rustpy._cg_rustpy", binding=Binding.PyO3, path="rust/Cargo.toml"
         )
     ],
     # rust extensions are not zip safe, just like C-extensions.

tests/test_project_points.py

@@ -9,7 +9,7 @@
 from cg.proj_jax import project_points_jax
 from cg.proj_cv import project_points_cv
 
-from cg_rustpy.cg_rustpy import  project_points_rs
+from cg_rustpy.proj_rs import project_points_rs
 
 
 DATA_DIR = Path(__file__).parent.parent / "data"