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>, } + impl $name { + pub fn inner(&self) -> &NodeState<'static, $value, DynamicGraph, DynamicGraph> { + self.inner.as_ref() + } + } + impl_node_state_ops!( $name, $value,