fix python bindings

kscalelabs · Oct 30, 2024 · e425f02 · e425f02
1 parent 4c0d317
commit e425f02
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 5 deletions.
diff --git a/pyklang/__init__.py b/pyklang/__init__.py
@@ -1,3 +1,5 @@
 """Defines the top-level package for PyKlang."""
 
 __version__ = "0.0.1"
+
+from .bindings import add
diff --git a/pyklang/pyproject.toml b/pyklang/pyproject.toml
@@ -5,6 +5,3 @@ build-backend = "maturin"
 [project]
 name = "bindings"
 requires-python = ">=3.9"
-
-[project.optional-dependencies]
-test = ["pytest", "pyright", "ruff"]
diff --git a/pyklang/src/lib.rs b/pyklang/src/lib.rs
@@ -9,7 +9,7 @@ fn add(a: u64, b: u64) -> u64 {
 }
 
 #[pymodule]
-fn pyklang(m: &Bound<PyModule>) -> PyResult<()> {
+fn bindings(m: &Bound<PyModule>) -> PyResult<()> {
     m.add_function(wrap_pyfunction!(add, m)?)?;
     Ok(())
 }

diff --git a/setup.py b/setup.py
@@ -49,7 +49,7 @@ def run(self) -> None:
     url="https://github.com/kscalelabs/klang",
     rust_extensions=[
         RustExtension(
-            target="pyklang",
+            target="pyklang.bindings",
             path="pyklang/Cargo.toml",
             binding=Binding.PyO3,
         ),