From 6a66b2a22c1df3c0915b05b61c95ccb0e96ac6cc Mon Sep 17 00:00:00 2001 From: timotheyca Date: Fri, 15 Jul 2022 21:38:21 +0300 Subject: [PATCH] ProtocolizedBinarySplit.protocolize --- .../trees/binary/core/protocolizedbinarysplit.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) 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(