diff --git a/rainbowadn/collection/trees/binary/core/protocolizedbinarysplit.py b/rainbowadn/collection/trees/binary/core/protocolizedbinarysplit.py index 3c7e174..864c088 100644 --- a/rainbowadn/collection/trees/binary/core/protocolizedbinarysplit.py +++ b/rainbowadn/collection/trees/binary/core/protocolizedbinarysplit.py @@ -26,14 +26,17 @@ class ProtocolizedBinarySplit( self.split = split self.tree = tree + def protocolize(self, tree: TreeType) -> BinaryProtocolized[ActiveKeyType, MetaDataType, TreeType]: + return BinaryProtocolized(self.protocol, tree) + def protocolized(self) -> BinaryProtocolized[ActiveKeyType, MetaDataType, TreeType]: - return BinaryProtocolized(self.protocol, self.tree) + return self.protocolize(self.tree) def protocolizedl(self) -> BinaryProtocolized[ActiveKeyType, MetaDataType, TreeType]: - return BinaryProtocolized(self.protocol, self.split.treel) + return self.protocolize(self.split.treel) def protocolizedr(self) -> BinaryProtocolized[ActiveKeyType, MetaDataType, TreeType]: - return BinaryProtocolized(self.protocol, self.split.treer) + return self.protocolize(self.split.treer) @classmethod async def split_of(