Sorted parser lib exports, fixed preprocess types

This commit is contained in:
Dmitriy Shishkov 2023-10-11 23:13:40 +03:00
parent 9cd21f717d
commit f3d24fe701
Signed by: dm1sh
GPG Key ID: 027994B0AA357688
2 changed files with 13 additions and 9 deletions

View File

@ -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",
)

View File

@ -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