diff --git a/pantab/_reader.py b/pantab/_reader.py index 32915724..36db20bd 100644 --- a/pantab/_reader.py +++ b/pantab/_reader.py @@ -1,7 +1,7 @@ import pathlib import shutil import tempfile -from typing import Dict, Optional, Union +from typing import Optional, Union import pandas as pd import pyarrow as pa @@ -27,9 +27,9 @@ def frame_from_hyper( def frames_from_hyper( source: Union[str, pathlib.Path], -) -> Dict[tab_api.TableName, pd.DataFrame]: +) -> dict[tab_api.TableName, pd.DataFrame]: """See api.rst for documentation.""" - result: Dict[TableType, pd.DataFrame] = {} + result: dict[TableType, pd.DataFrame] = {} table_names = [] with tempfile.TemporaryDirectory() as tmp_dir, tab_api.HyperProcess( diff --git a/pantab/_writer.py b/pantab/_writer.py index 0068979b..f6b0b296 100644 --- a/pantab/_writer.py +++ b/pantab/_writer.py @@ -2,7 +2,7 @@ import shutil import tempfile import uuid -from typing import Dict, Optional, Union +from typing import Optional, Union import pandas as pd import pyarrow as pa @@ -33,7 +33,7 @@ def frame_to_hyper( def frames_to_hyper( - dict_of_frames: Dict[pantab_types.TableType, pd.DataFrame], + dict_of_frames: dict[pantab_types.TableType, pd.DataFrame], database: Union[str, pathlib.Path], table_mode: str = "w", *,