Source code for pd_utils.optimize.load

import pandas as pd

from pd_utils.optimize.typing import DfOrSeries


[docs]def read_file(filepath: str, **read_func_kwargs) -> DfOrSeries: extension = filepath.rpartition(".")[-1].lower() if extension == "csv": df = pd.read_csv(filepath, **read_func_kwargs) # TODO [#4]: in read_file, determine filetype and use proper loader # # Currently it just handles csv else: raise NotImplementedError(f"could not load filetype {extension}") # TODO [#5]: make optimize df an option for read_file #### TEMP # from pd_utils.optimize.dftypes import optimized_df # df = optimized_df(df) ### END TEMP return df