From c340a1b8243da52c70dd2c6f05cddfdaf336c82e Mon Sep 17 00:00:00 2001
From: ljeub-pometry <97447091+ljeub-pometry@users.noreply.github.com>
Date: Mon, 2 Dec 2024 11:27:51 +0100
Subject: [PATCH] make the inner rust value accessible on python NodeState and
 LazyNodeState wrappers (#1878)

make the inner rust value accessible on python NodeState and LazyNodeState wrappers so they can be used as input arguments for other functions
---
 raphtory/src/python/graph/node_state.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/raphtory/src/python/graph/node_state.rs b/raphtory/src/python/graph/node_state.rs
index f2162b30c..4fe0144d6 100644
--- a/raphtory/src/python/graph/node_state.rs
+++ b/raphtory/src/python/graph/node_state.rs
@@ -255,6 +255,12 @@ macro_rules! impl_lazy_node_state {
             inner: LazyNodeState<'static, $op, DynamicGraph, DynamicGraph>,
         }
 
+        impl $name {
+            pub fn inner(&self) -> &LazyNodeState<'static, $op, DynamicGraph, DynamicGraph> {
+                &self.inner
+            }
+        }
+
         #[pymethods]
         impl $name {
             /// Compute all values and return the result as a node view
@@ -306,6 +312,12 @@ macro_rules! impl_node_state {
             inner: Arc<NodeState<'static, $value, DynamicGraph, DynamicGraph>>,
         }
 
+        impl $name {
+            pub fn inner(&self) -> &NodeState<'static, $value, DynamicGraph, DynamicGraph> {
+                self.inner.as_ref()
+            }
+        }
+
         impl_node_state_ops!(
             $name,
             $value,