create python NodeState from Arc

Pometry · Dec 9, 2024 · 9eb475d · 9eb475d
1 parent 419db76
commit 9eb475d
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/raphtory/src/python/graph/node_state.rs b/raphtory/src/python/graph/node_state.rs
@@ -313,8 +313,8 @@ macro_rules! impl_node_state {
         }
 
         impl $name {
-            pub fn inner(&self) -> &NodeState<'static, $value, DynamicGraph, DynamicGraph> {
-                self.inner.as_ref()
+            pub fn inner(&self) -> &Arc<NodeState<'static, $value, DynamicGraph, DynamicGraph>> {
+                &self.inner
             }
         }
 
@@ -335,6 +335,12 @@ macro_rules! impl_node_state {
             }
         }
 
+        impl From<Arc<NodeState<'static, $value, DynamicGraph, DynamicGraph>>> for $name {
+            fn from(inner: Arc<NodeState<'static, $value, DynamicGraph, DynamicGraph>>) -> Self {
+                $name { inner }
+            }
+        }
+
         impl pyo3::IntoPy<PyObject> for NodeState<'static, $value, DynamicGraph, DynamicGraph> {
             fn into_py(self, py: Python<'_>) -> PyObject {
                 $name::from(self).into_py(py)