def extract_data(spark, schema):
    """Load data from csv file format. In production, data will be read from SQL db

    :param spark: Spark session object.
    :return: Spark DataFrame.
    """
    df = (
        spark.read.format('csv')
        .option('sep', ',')
        .option('header', 'true')
        .schema(schema)
        .load('./data/raw_data/rawdata.csv')
    )

    return df