From faa96816f9c84f55b980550ccd1d6ab5f407a31f Mon Sep 17 00:00:00 2001 From: timotheyca Date: Sun, 11 Sep 2022 17:11:26 +0300 Subject: [PATCH] minor fix --- rainbowadn/collection/trees/binary/core/__init__.py | 2 ++ rainbowadn/collection/trees/binary/core/binarycreation.py | 6 +++--- rainbowadn/core/gather.py | 4 ++-- 3 files changed, 7 insertions(+), 5 deletions(-) diff --git a/rainbowadn/collection/trees/binary/core/__init__.py b/rainbowadn/collection/trees/binary/core/__init__.py index 9afa057..d6cf214 100644 --- a/rainbowadn/collection/trees/binary/core/__init__.py +++ b/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 diff --git a/rainbowadn/collection/trees/binary/core/binarycreation.py b/rainbowadn/collection/trees/binary/core/binarycreation.py index 02f4498..a9cd1e3 100644 --- a/rainbowadn/collection/trees/binary/core/binarycreation.py +++ b/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] diff --git a/rainbowadn/core/gather.py b/rainbowadn/core/gather.py index 04d2721..30467db 100644 --- a/rainbowadn/core/gather.py +++ b/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):