src/data/utils.py

import pandas as pd
data_set_options = [
    'Airline',
    'Milk production'
]

sample_or_upload_options = [
    'Use sample data',
    'Upload data'
]


def import_sample_data(sample_data_selected, data_set_options):
    if sample_data_selected == data_set_options[0]:
        data = pd.read_csv('data/processed/airline.csv',
                           parse_dates=['Month'], index_col='Month')

    if sample_data_selected == data_set_options[1]:
        data = pd.read_csv('data/processed/milk_production.csv',
                           parse_dates=['date'], index_col='date')

    return data