import pandas as pd
from importlib import resources
[docs]
def load_soccer_data():
"""
Load the soccer data from a parquet file included in the package.
Returns
-------
pandas.DataFrame
The loaded soccer data.
"""
# Use 'importlib.resources' to ensure the path is correct after installation
data_path = resources.files("stepsel.data").joinpath("soccer.parquet")
dt = pd.read_parquet(data_path)
return dt
[docs]
def load_dummy_data():
"""
Load the dummy data from a parquet file included in the package.
Returns
-------
pandas.DataFrame
The loaded dummy data.
"""
# Use 'importlib.resources' to ensure the path is correct after installation
data_path = resources.files("stepsel.data").joinpath("dummy_data.parquet")
dt = pd.read_parquet(data_path)
return dt