diff --git a/pandas/core/internals/managers.py b/pandas/core/internals/managers.py index a1e5468e2f871..907498c7ff350 100644 --- a/pandas/core/internals/managers.py +++ b/pandas/core/internals/managers.py @@ -3,7 +3,7 @@ import itertools import operator import re -from typing import List, Optional, Union +from typing import List, Optional, Sequence, Tuple, Union import numpy as np @@ -95,9 +95,12 @@ class BlockManager(PandasObject): __slots__ = ['axes', 'blocks', '_ndim', '_shape', '_known_consolidated', '_is_consolidated', '_blknos', '_blklocs'] - def __init__(self, blocks, axes, do_integrity_check=True): + def __init__(self, + blocks: Sequence[Block], + axes: Sequence[Index], + do_integrity_check: bool = True): self.axes = [ensure_index(ax) for ax in axes] - self.blocks = tuple(blocks) + self.blocks = tuple(blocks) # type: Tuple[Block, ...] for block in blocks: if block.is_sparse: @@ -1415,8 +1418,11 @@ class SingleBlockManager(BlockManager): _known_consolidated = True __slots__ = () - def __init__(self, block, axis, do_integrity_check=False, fastpath=False): - + def __init__(self, + block: Block, + axis: Union[Index, List[Index]], + do_integrity_check: bool = False, + fastpath: bool = False): if isinstance(axis, list): if len(axis) != 1: raise ValueError("cannot create SingleBlockManager with more " @@ -1455,7 +1461,7 @@ def __init__(self, block, axis, do_integrity_check=False, fastpath=False): if not isinstance(block, Block): block = make_block(block, placement=slice(0, len(axis)), ndim=1) - self.blocks = [block] + self.blocks = tuple([block]) def _post_setstate(self): pass