diff --git a/redframes/types.py b/redframes/types.py index ca2d34e..23a83c6 100644 --- a/redframes/types.py +++ b/redframes/types.py @@ -1,18 +1,18 @@ from __future__ import annotations import datetime -from typing import Any, Callable, Literal, Union +from typing import Any, Callable, Literal, Union, List import numpy as np # pyright: ignore[reportMissingImports] import pandas as pd # pyright: ignore[reportMissingImports] import pandas.core.groupby.generic as pg # pyright: ignore[reportMissingImports] Value = Any -Values = list[Value] +Values = List[Value] OldValue = Value NewValue = Value Column = str -Columns = list[Column] +Columns = List[Column] LazyColumns = Union[Column, Columns] OldColumn = Column NewColumn = Column diff --git a/redframes/verbs/rename.py b/redframes/verbs/rename.py index 7796895..2195494 100644 --- a/redframes/verbs/rename.py +++ b/redframes/verbs/rename.py @@ -1,8 +1,10 @@ +from typing import Dict + from ..checks import _check_type, _check_values from ..types import NewColumn, OldColumn, PandasDataFrame -def rename(df: PandasDataFrame, columns: dict[OldColumn, NewColumn]) -> PandasDataFrame: +def rename(df: PandasDataFrame, columns: Dict[OldColumn, NewColumn]) -> PandasDataFrame: _check_type(columns, dict) cv = columns.values() _check_values(cv, str) diff --git a/redframes/verbs/replace.py b/redframes/verbs/replace.py index c2e250e..dce48e8 100644 --- a/redframes/verbs/replace.py +++ b/redframes/verbs/replace.py @@ -1,9 +1,11 @@ +from typing import Dict + from ..checks import _check_type from ..types import Column, NewValue, OldValue, PandasDataFrame def replace( - df: PandasDataFrame, over: dict[Column, dict[OldValue, NewValue]] + df: PandasDataFrame, over: Dict[Column, Dict[OldValue, NewValue]] ) -> PandasDataFrame: _check_type(over, dict) bad_columns = list(set(over.keys()) - set(df.columns))