minor fix

rainbowadn/collection/trees/binary/core/__init__.py

@@ -2,6 +2,7 @@ __all__ = (
     'BalancedCreation',
     'BinaryBalancing',
     'BinaryCreation',
+    'BinaryMetadata',
     'BinaryProtocolized',
     'BinarySplit',
     'ProtocolizedBinarySplit',
@@ -10,6 +11,7 @@ __all__ = (
 from .balancedcreation import BalancedCreation
 from .binarybalancing import BinaryBalancing
 from .binarycreation import BinaryCreation
+from .binarymetadata import BinaryMetadata
 from .binaryprotocolized import BinaryProtocolized
 from .binarysplit import BinarySplit
 from .protocolizedbinarysplit import ProtocolizedBinarySplit

rainbowadn/collection/trees/binary/core/binarycreation.py

@@ -12,9 +12,9 @@ MetaDataType = TypeVar('MetaDataType')
 
 
 class BinaryCreation(Generic[ActiveKeyType, MetaDataType, TreeType]):
-    def __init__(self, comparator_: Comparator[ActiveKeyType]):
-        assert isinstance(comparator_, Comparator)
-        self.comparator = comparator_
+    def __init__(self, comparator: Comparator[ActiveKeyType]):
+        assert isinstance(comparator, Comparator)
+        self.comparator = comparator
 
     async def split(self, tree: TreeType) -> Optional[
         BinarySplit[ActiveKeyType, MetaDataType, TreeType]

rainbowadn/core/gather.py

@@ -61,8 +61,8 @@ def gather(
 ) -> Awaitable[tuple[T0, T1, T2, T3, T4]]: ...
 
 
-@overload
-def gather(*args: Coroutine[Any, Any, T0]) -> Awaitable[tuple[T0, ...]]: ...
+# @overload
+# def gather(*args: Coroutine[Any, Any, T0]) -> Awaitable[tuple[T0, ...]]: ...
 
 
 def gather(*args):