pyo3

ComputationalPsychiatry · Oct 14, 2024 · e7c37a9 · e7c37a9
1 parent 7b8e62f
commit e7c37a9
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 31 deletions.
diff --git a/src/hgf/Cargo.lock b/src/hgf/Cargo.lock
diff --git a/src/hgf/src/network.rs b/src/hgf/src/network.rs
@@ -1,7 +1,6 @@
 use std::collections::HashMap;
 use crate::updates::posterior;
 use pyo3::prelude::*;
-use pyo3::wrap_pyfunction;
 
 #[derive(Debug)]
 pub struct AdjacencyLists{
@@ -144,8 +143,8 @@ impl Network {
 
 // Create a module to expose the class to Python
 #[pymodule]
-fn my_rust_library(_py: Python, m: &PyModule) -> PyResult<()> {
-    m.add_class::<Network>()?; // Add the class to the Python module
+fn my_module(m: &Bound<'_, PyModule>) -> PyResult<()> {
+    m.add_class::<Network>()?;
     Ok(())
 }