diff --git a/parser/__init__.py b/parser/__init__.py index 0502502..cd4b8b3 100644 --- a/parser/__init__.py +++ b/parser/__init__.py @@ -1,10 +1,12 @@ from .address import split_addresses from .building_id import ( + async_fetch_building_id, async_fetch_building_ids, concurrent_fetch_builing_ids, fetch_builing_ids, get_building_id, ) +from .lenenergo import LenenergoParser from .preprocess import ( COL_NS, ICOL_NS, @@ -12,20 +14,20 @@ from .preprocess import ( preprocess_df, preprocess_read_df, ) -from .lenenergo import LenenergoParser from .util import pipeline __all__ = ( - "LenenergoParser", - "split_addresses", - "get_building_id", - "fetch_builing_ids", + "async_fetch_building_id", "async_fetch_building_ids", - "concurrent_fetch_builing_ids", - "preprocess_df", "COL_NS", - "ICOL_NS", - "preprocess_read_df", + "concurrent_fetch_builing_ids", + "fetch_builing_ids", + "get_building_id", "group_by_index", + "ICOL_NS", + "LenenergoParser", "pipeline", + "preprocess_df", + "preprocess_read_df", + "split_addresses", ) diff --git a/parser/preprocess.py b/parser/preprocess.py index 61fa8d6..5de5c98 100644 --- a/parser/preprocess.py +++ b/parser/preprocess.py @@ -34,6 +34,8 @@ def preprocess_df(df: pd.DataFrame) -> pd.DataFrame: ) df.drop(columns=[f"{a}_date", f"{a}_time"], inplace=True) + df = df.convert_dtypes() + return df